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