Update app.py

app.py

@@ -71,7 +71,7 @@ DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 EFFICIENT_SAM_MODEL = load(device=DEVICE)
 YOLO_WORLD_MODEL = YOLOWorld(model_id="yolo_world/l")
 
-BOUNDING_BOX_ANNOTATOR = sv.BoundingBoxAnnotator()
+BOX_ANNOTATOR = sv.BoxAnnotator()
 MASK_ANNOTATOR = sv.MaskAnnotator()
 LABEL_ANNOTATOR = sv.LabelAnnotator()
 
@@ -99,7 +99,7 @@ def annotate_image(
         zip(detections.class_id, detections.confidence)
     ]
     output_image = MASK_ANNOTATOR.annotate(input_image, detections)
-    output_image = BOUNDING_BOX_ANNOTATOR.annotate(output_image, detections)
+    output_image = BOX_ANNOTATOR.annotate(output_image, detections)
     output_image = LABEL_ANNOTATOR.annotate(output_image, detections, labels=labels)
     return output_image