Update app.py

app.py

@@ -5,19 +5,23 @@ import plotly.express as px
 import numpy as np
 import random
 from ultralytics import YOLO
-from sahi.models.yolov8 import *
+#from sahi.models.yolov8 import *
+from src.sahi_onnx import *
 from sahi.predict import get_sliced_prediction
 from sahi.utils.cv import visualize_object_predictions
 import PIL
 
-model_base = "https://huggingface.co/mayrajeo/marine-vessel-detection/resolve/main/"
+#model_base = "https://huggingface.co/mayrajeo/marine-vessel-detection/resolve/main/"
+model_base = 'onnx_models'
 
 def inference(
     im:gr.inputs.Image=None,
     model_path:gr.inputs.Dropdown='YOLOv8n',
     conf_thr:gr.inputs.Slider=0.25
 ):
-    model = Yolov8DetectionModel(model_path=f'{model_base}/{model_path}/{model_path}.pt',
+    #model = Yolov8DetectionModel(model_path=f'{model_base}/{model_path}/{model_path}.pt',
+    model = Yolov8onnxDetectionModel(model_path=f'{model_base}/{model_path}/best.onnx',
+                                     config_path=f'{model_base}/{model_path}/args.yaml''
                                  device='cpu',
                                  confidence_threshold=conf_thr,
                                  image_size=640)