Update app.py

app.py

@@ -32,8 +32,8 @@ model = YOLO('best (1).pt')
 def response(image):
     res = Image.open(image)
     im_rgb = model(res)
-    im_rgb = Image.fromarray(im_rgb)
-    return im_rgb
+    # im_rgb = Image.fromarray(im_rgb)
+    return np.array(im_rgb)
 
 iface = gr.Interface(fn=response, inputs="image", outputs="image")
 iface.launch()