AlshimaaGamalAlsaied commited on
Commit
b276cff
·
1 Parent(s): 4a681b8
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ from huggingface_hub import hf_hub_download
11
 
12
  def yolov7_inference(
13
  image: gr.inputs.Image = None,
14
- model_path: gr.inputs.Dropdown = hf_hub_download(repo_id=f"model_baseline", filename=f"model_baseline/tree/main/best_baseline.pt"),
15
  image_size: gr.inputs.Slider = 640,
16
  conf_threshold: gr.inputs.Slider = 0.25,
17
  iou_threshold: gr.inputs.Slider = 0.45,
 
11
 
12
  def yolov7_inference(
13
  image: gr.inputs.Image = None,
14
+ model_path: gr.inputs.Dropdown = hf_hub_download(repo_id=f"model_baseline", filename=f"best%20_baseline.pt"),
15
  image_size: gr.inputs.Slider = 640,
16
  conf_threshold: gr.inputs.Slider = 0.25,
17
  iou_threshold: gr.inputs.Slider = 0.45,