update

app.py

@@ -3,16 +3,18 @@ import gradio as gr
 import yolov7
 
 
-from huggingface_hub import hf_hub_download
+#
+# from huggingface_hub import hf_hub_download
 from huggingface_hub import hfapi
 
+
 # Images
 #torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg', 'zidane.jpg')
 #torch.hub.download_url_to_file('https://raw.githubusercontent.com/obss/sahi/main/tests/data/small-vehicles1.jpeg', 'small-vehicles1.jpeg')
     
 def yolov7_inference(
     image: gr.inputs.Image = None,
-    model_path: gr.inputs.Dropdown = hfapi(repo_id=f"model_baseline", filename=f"best_baseline.pt"),
+    model_path: gr.inputs.Dropdown = hfapi(repo_id=f"alshimaa/model_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,

requirements.txt

@@ -1,3 +1,4 @@
 gradio
 torch
-yolov7detect
+yolov7detect
+hfapi