Update app.py

app.py

@@ -8,7 +8,6 @@ from transformers import DetrImageProcessor, DetrForObjectDetection
 processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50", revision="no_timm")
 model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50", revision="no_timm")
 
-# Define the object detection function
 def detect_objects(frame):
     # Convert the frame to PIL image
     image = Image.fromarray(frame)
@@ -28,7 +27,8 @@ def detect_objects(frame):
         box = [round(i, 2) for i in box.tolist()]
         frame = gr.draw_box(frame, box, label=model.config.id2label[label.item()], color=(0, 255, 0))
 
-    return frame
+    # Convert frame back to numpy array for Gradio
+    return np.array(frame)
 
 # Define the Gradio interface
 iface = gr.Interface(