update

app.py

@@ -11,7 +11,7 @@ from huggingface_hub import hf_hub_download
     
 def yolov7_inference(
     image: gr.inputs.Image = None,
-    model_path: gr.inputs.Dropdown = None,
+    model_path: gr.inputs.Dropdown = hf_hub_download(repo_id=f"{best_baseline}", filename=f"{best_baseline}.pt"),
     image_size: gr.inputs.Slider = 640,
     conf_threshold: gr.inputs.Slider = 0.25,
     iou_threshold: gr.inputs.Slider = 0.45,
@@ -39,10 +39,10 @@ inputs = [
     gr.inputs.Image(type="pil", label="Input Image"),
     gr.inputs.Dropdown(
         choices=[
-            "best_baseline",
+            "kadirnar/yolov7-tiny-v0.1",
             "kadirnar/yolov7-v0.1",
         ],
-        default="best_baseline",
+        default="kadirnar/yolov7-tiny-v0.1",
         label="Model",
     ),
     gr.inputs.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),