fix: 🐞 Label and Box Annotators fixed

Signed-off-by: Onuralp SEZER <[email protected]>

app.py

@@ -4,7 +4,8 @@ from ultralytics import YOLO
 import spaces
 import supervision as sv
 
-box_annotator = sv.BoxAnnotator()
+BOX_ANNOTATOR = sv.BoxAnnotator()
+LABEL_ANNOTATOR = sv.LabelAnnotator()
 
 category_dict = {
     0: 'person', 1: 'bicycle', 2: 'car', 3: 'motorcycle', 4: 'airplane', 5: 'bus',
@@ -56,7 +57,12 @@ def LeYOLO_inference(image, model_id, image_size, conf_threshold, iou_threshold)
         f"{category_dict[class_id]} {confidence:.2f}"
         for class_id, confidence in zip(detections.class_id, detections.confidence)
     ]
-    annotated_image = box_annotator.annotate(image, detections=detections, labels=labels)
+    
+
+    annotated_image = BOX_ANNOTATOR.annotate(
+        scene=image, detections=detections)
+    annotated_image = LABEL_ANNOTATOR.annotate(
+        scene=annotated_image, detections=detections, labels=labels)
 
     return annotated_image