mayrajeo commited on
Commit
5edd9d1
1 Parent(s): 5071d43

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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}/best.onnx',
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,