Update app.py

app.py

@@ -9,20 +9,14 @@ model = None
 def predict_image(img):
     """Predicts objects in an image using a YOLOv8 model with adjustable confidence and IOU thresholds."""
     model = YOLO("best")
-    results = model.predict(
-        source=img,
-        conf=0.25,
-        iou=0.45,
-        show_labels=True,
-        show_conf=True,
-        imgsz=640,
-    )
-
-    for r in results:
-        im_array = r.plot()
-        im = Image.fromarray(im_array[..., ::-1])
-
-    return im
+    model.predict(source=img, save=True)
+    results = model(img)
+
+    # for r in results:
+    #     im_array = r.plot()
+    #     im = Image.fromarray(im_array[..., ::-1])
+    
+    return results[0]
 
 
 demo = gr.Interface(