Update app.py

app.py

@@ -25,6 +25,18 @@ modin=f"yolov{ver[2]}{ltr[1]}-{tsk[0]}.pt"
 print(modin)
 model = YOLO(modin)
 
+def draw_box(image,det):
+    height, width, channels = image.shape
+    for ea in det.xyxy:
+        #bbox = convert_coords(ea, width, height)
+        #print(bbox)
+        start_point = ((int(ea[0]),int(ea[1])))
+        end_point = ((int(ea[2]),int(ea[3])))
+        color = (255, 0, 0)
+        thickness = 2
+        image = cv2.rectangle(image, start_point, end_point, color, thickness)
+    return image
+
 @spaces.GPU
 def stream_object_detection(video):
     SUBSAMPLE=1
@@ -66,5 +78,7 @@ with gr.Blocks() as app:
             inp = gr.Video()
             btn = gr.Button()
         outp_v = gr.Video(label="Processed Video", streaming=True, autoplay=True)
-    btn.click(stream_object_detection,inp,[outp_v])
+    outp_j = gr.JSON()
+
+    btn.click(stream_object_detection,inp,[outp_v,outp_j])
 app.queue().launch()