Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import sahi
 import torch
-from ultralyticsplus import YOLO
+from ultralytics import YOLO
 
 # Download images
 
@@ -42,9 +42,10 @@ def yolov8_inference(
         current_model_name = model_name
     model.overrides["conf"] = conf_threshold
     model.overrides["iou"] = iou_threshold
-    results = model.predict(image, imgsz=image_size)
-
+    results = model(image, return_outputs=True)
+    print(results)
     outputs=[]
+
     for i,box in enumerate(results[0].boxes):
         label = results[0].names[box.cls[0].item()]
         bbox = box.xyxy[0]