Update app.py

app.py

@@ -26,13 +26,9 @@ def detect(img):
     boxes = predictions[:, :4] # x1, y1, x2, y2
     scores = predictions[:, 4]
     categories = predictions[:, 5]
-    img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
-    for i, box in enumerate(boxes):
-        cv2.rectangle(img, tuple(box[:2]), tuple(box[2:]), (0, 255, 0), 2)
 
     # save image as numpy array
-    img_array = np.array(img)
-    return image_array
+    return results.numpy()
 # show detection bounding boxes on image
 
 img = gr.inputs.Image(shape=(192, 192))