Update app.py

app.py

@@ -30,10 +30,14 @@ def image_preprocess(image):
 model = YOLO('best (1).pt')
 
 def response(image):
-    res = Image.fromarray(image)
-    im_rgb = model(res)
+    results = model(image)
+    for i, r in enumerate(results):
+    # Plot results image
+      im_bgr = r.plot()  # BGR-order numpy array
+      im_rgb = im_bgr[..., ::-1]  # Convert BGR to RGB
+    
     # im_rgb = Image.fromarray(im_rgb)
-    return np.array(im_rgb)
+    return im_rgb
 
 iface = gr.Interface(fn=response, inputs="image", outputs="image")
 iface.launch()