Update app.py

app.py

@@ -40,12 +40,19 @@ def process_image(image: PIL.Image.Image) -> tuple[PIL.Image.Image, list[dict]]:
     )
     result = results[0]  # take first image results
     boxes_xyxy = result["boxes"].cpu().numpy()
+    indexes = result["labels"].cpu().numpy()
+    scores = result["scores"].cpu().numpy()
+    text_labels = [
+        f"{model.id2label[index]} [{score.item():.2f}]" for index, score in zip(indexes, scores)
+    ]
 
     detections = sv.Detections(xyxy=boxes_xyxy)
     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)
+    annotated_frame = label_annotator(scene=annotated_frame, detections=detections, labels=text_labels)
 
     return annotated_frame