Update app.py

app.py

@@ -12,7 +12,7 @@ model = DetrForObjectDetection.from_pretrained("Guy2/AirportSec-150epoch")
 id2label = {0: 'dangerous-items', 1: 'Gun', 2: 'Knife', 3: 'Pliers', 4: 'Scissors', 5: 'Wrench'}
 def anylize(url):
     image = Image.open(requests.get(url, stream=True).raw)
-    return image
+    # return image
     with torch.no_grad():
 
         inputs = image_processor(images=image, return_tensors='pt')
@@ -29,8 +29,9 @@ 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
+    return labels
+    # json_list = json.dumps(labels)
+    # return json_list
 
-# gr.Interface(fn = anylize, inputs="text", outputs=gr.JSON()).launch()
-gr.Interface(fn = anylize, inputs="text", outputs="image").launch()
+gr.Interface(fn = anylize, inputs="text", outputs="text").launch()
+# gr.Interface(fn = anylize, inputs="text", outputs="image").launch()