AlshimaaGamalAlsaied commited on
Commit
c824d7a
1 Parent(s): dfe8c33
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -14,7 +14,7 @@ from huggingface_hub import HfApi
14
 
15
  def yolov7_inference(
16
  image: gr.inputs.Image = None,
17
- model_path: gr.inputs.Dropdown = HfApi(repo_id=f"alshimaa/model_baseline", filename=f"best_baseline.pt"),
18
  image_size: gr.inputs.Slider = 640,
19
  conf_threshold: gr.inputs.Slider = 0.25,
20
  iou_threshold: gr.inputs.Slider = 0.45,
@@ -42,10 +42,11 @@ inputs = [
42
  gr.inputs.Image(type="pil", label="Input Image"),
43
  gr.inputs.Dropdown(
44
  choices=[
 
45
  "kadirnar/yolov7-tiny-v0.1",
46
  "kadirnar/yolov7-v0.1",
47
  ],
48
- default="kadirnar/yolov7-tiny-v0.1",
49
  label="Model",
50
  ),
51
  gr.inputs.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),
 
14
 
15
  def yolov7_inference(
16
  image: gr.inputs.Image = None,
17
+ model_path: gr.inputs.Dropdown = None,
18
  image_size: gr.inputs.Slider = 640,
19
  conf_threshold: gr.inputs.Slider = 0.25,
20
  iou_threshold: gr.inputs.Slider = 0.45,
 
42
  gr.inputs.Image(type="pil", label="Input Image"),
43
  gr.inputs.Dropdown(
44
  choices=[
45
+ "alshimaa/model_baseline"
46
  "kadirnar/yolov7-tiny-v0.1",
47
  "kadirnar/yolov7-v0.1",
48
  ],
49
+ default="alshimaa/model_baseline",
50
  label="Model",
51
  ),
52
  gr.inputs.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),