Update app.py

app.py

@@ -26,7 +26,13 @@ def detect(img):
     boxes = predictions[:, :4] # x1, y1, x2, y2
     scores = predictions[:, 4]
     categories = predictions[:, 5]
-    return results.show()
+    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
 # show detection bounding boxes on image
 
 img = gr.inputs.Image(shape=(192, 192))