Update app.py

app.py

@@ -28,9 +28,9 @@ def predict(image):
     labels = [f"{id2label[class_id]} {confidence:.2f}" for _, confidence, class_id, _ in detections]
     frame = box_annotator.annotate(scene=image.copy(), detections=detections, labels=labels)
     
-    return plot_image(frame, (16, 16))
+    #return plot_image(frame, (16, 16))
     
-    #return labels
+    return frame
 
 gr.Interface(
     predict,