AlshimaaGamalAlsaied commited on
Commit
32f1f74
·
1 Parent(s): 3e5f5c4
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"{best_baseline}", filename=f"{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"best_baseline", filename=f"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,