Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ def inference(
|
|
20 |
conf_thr:gr.inputs.Slider=0.25
|
21 |
):
|
22 |
#model = Yolov8DetectionModel(model_path=f'{model_base}/{model_path}/{model_path}.pt',
|
23 |
-
model = Yolov8onnxDetectionModel(model_path=f'{model_base}/{model_path}/
|
24 |
config_path=f'{model_base}/{model_path}/args.yaml',
|
25 |
device='cpu',
|
26 |
confidence_threshold=conf_thr,
|
|
|
20 |
conf_thr:gr.inputs.Slider=0.25
|
21 |
):
|
22 |
#model = Yolov8DetectionModel(model_path=f'{model_base}/{model_path}/{model_path}.pt',
|
23 |
+
model = Yolov8onnxDetectionModel(model_path=f'{model_base}/{model_path}/{model_path.lower()}.onnx',
|
24 |
config_path=f'{model_base}/{model_path}/args.yaml',
|
25 |
device='cpu',
|
26 |
confidence_threshold=conf_thr,
|