Update app.py

app.py

@@ -23,7 +23,7 @@ def detect(img):
     boxes = predictions[:, :4] # x1, y1, x2, y2
     scores = predictions[:, 4]
     categories = predictions[:, 5]
-    return cv2.imread(results.save())
+    annotated_image = np.squeeze(results.render())
 
 
 img = gr.inputs.Image(shape=(192, 192))