Update app.py

app.py

@@ -26,8 +26,10 @@ def anylize(url):
 
     # annotate
     detections = sv.Detections.from_transformers(transformers_results=results).with_nms(threshold=0.5)
-    labels = [str([list(xyxy), confidence, id2label[class_id]]) for xyxy, _, confidence, class_id, _ in detections]
-    json_list = json.dumps(labels)
-    return json_list
+    labels = [f"{id2label[class_id]} {confidence:.2f}" for _, _, confidence, class_id, _ in detections]
+    box_annotator = sv.BoxAnnotator()
+    frame = box_annotator.annotate(scene=image.copy(), detections=detections, labels=labels)
+    return frame
 
-gr.Interface(fn = anylize, inputs="text", outputs=gr.JSON()).launch()
+output = gr.components.Image(type="numpy", label="Output Image")
+gr.Interface(fn = anylize, inputs="text", outputs=output).launch()