Update app.py

app.py

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