Update app.py

app.py

@@ -46,12 +46,12 @@ def process_image(image: PIL.Image.Image) -> tuple[PIL.Image.Image, list[dict]]:
         f"{model.config.id2label[index]} [{score.item():.2f}]" for index, score in zip(indexes, scores)
     ]
 
-    detections = sv.Detections(xyxy=boxes_xyxy)
+    detections = sv.Detections(xyxy=boxes_xyxy, class_id=indexes, confidence=scores)
     bounding_box_annotator = sv.BoxAnnotator(color=sv.Color.WHITE, color_lookup=sv.ColorLookup.INDEX, thickness=1)
     label_annotator = sv.LabelAnnotator()
 
     # annotate bounding boxes
-    annotated_frame = bounding_box_annotator.annotate(scene=image.copy(), detections=detections, class_id=indexes, confidence=scores)
+    annotated_frame = bounding_box_annotator.annotate(scene=image.copy(), detections=detections)
     annotated_frame = label_annotator.annotate(scene=annotated_frame, detections=detections, labels=text_labels)
 
     return annotated_frame