Update app.py

app.py

@@ -6,10 +6,14 @@ from ultralytics import YOLO
 model = YOLO('best (1).pt')
 
 def response(image):
-    img = Image.fromarray(image)
-    results = model(img)
+    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
+    plt.figure(figsize=(10, 10))
+    plt.imshow(im_rgb)
+    plt.axis('off')
+    plt.show()
     return results
 
 iface = gr.Interface(fn=response, inputs="image", outputs="image")