update

app.py

@@ -2,19 +2,17 @@ import gradio as gr
 from transformers import DetrImageProcessor, DetrForObjectDetection
 import torch
 import supervision as sv
+import json
 
 def anylize(img):
-    id2label = {k: v['name'] for k,v in categories.items()}
-    box_annotator = sv.BoxAnnotator()
+    id2label = {0: 'dangerous-items', 1: 'Gun', 2: 'Knife', 3: 'Pliers', 4: 'Scissors', 5: 'Wrench'}
     image = img
 
     with torch.no_grad():
 
-        # load image and predict
         inputs = image_processor(images=image, return_tensors='pt')
         outputs = model(**inputs)
     
-        # post-process
         target_sizes = torch.tensor([image.shape[:2]])
         results = image_processor.post_process_object_detection(
             outputs=outputs, 
@@ -24,7 +22,7 @@ def anylize(img):
 
     # annotate
     detections = sv.Detections.from_transformers(transformers_results=results).with_nms(threshold=0.5)
-    labels = [f"{id2label[class_id]} {confidence:.2f}" for _, confidence, class_id, _ in detections]
-    frame = box_annotator.annotate(scene=image.copy(), detections=detections, labels=labels)
+    labels = [str([list(xyxy), confidence, id2label[class_id]]) for xyxy, _, confidence, class_id, _ in detections]
+    json_list = json.dumps(str(labels[0]))
 
 gr.Interface.load("models/Guy2/AirportSec-100epoch").launch()