Update app.py

app.py

@@ -5,17 +5,20 @@ import PIL.Image as Image
 from ultralytics import YOLO, YOLOv10
 from huggingface_hub import hf_hub_download, HfApi
 
+global repo_id="atalaydenknalbant/asl-models"
+
 def download_models(model_id):
-    hf_hub_download("atalaydenknalbant/asl-models", filename=f"{model_id}", local_dir=f"./")
+    hf_hub_download(repo_id, filename=f"{model_id}", local_dir=f"./")
     return f"./{model_id}"
 
-def get_model_filenames(repo_id="atalaydenknalbant/asl-models", file_extension=".pt"):
+def get_model_filenames(repo_id, file_extension=".pt"):
     api = HfApi()
     files = api.list_repo_files(repo_id)
     model_filenames = [file for file in files if file.endswith(file_extension)]
     return model_filenames
 
-model_filenames = get_model_filenames("atalaydenknalbant/asl-models")
+
+model_filenames = get_model_filenames(repo_id)
 print("Model filenames:", model_filenames)
 
 box_annotator = sv.BoxAnnotator()
@@ -49,18 +52,7 @@ def app():
         with gr.Row():
             with gr.Column():
                 image = gr.Image(type="pil", label="Image", interactive=True)
-                
-                # model_id = gr.Dropdown(
-                #     label="Model",
-                #     choices=[
-                #         "yolov10x.pt",
-                #         "yolov10s.pt",
-                #         "yolov9e.pt",
-                #         "yolov9s.pt",
-                #         "yolov8x.pt",
-                #     ],
-                #     value="yolov10s.pt",
-                # )
+    
                 model_id = gr.Dropdown(
                     label="Model",
                     choices=model_filenames,