Update app.py

app.py

@@ -2,22 +2,21 @@ import spaces
 import supervision as sv
 import PIL.Image as Image
 from ultralytics import YOLO
-from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download, HfApi
 import gradio as gr
 import torch
 global repo_id
 
 repo_id = "atalaydenknalbant/asl-yolo-models"
 
+def get_model_filenames(repo_id):
+    api = HfApi()
+    files = api.list_repo_files(repo_id)
+    model_filenames = [file for file in files if file.endswith('.pt')]
+    return model_filenames
 
-model_filenames = [
-    "yolov10s.pt",
-    "yolov10x.pt",
-    "yolov8s.pt",
-    "yolov8x.pt",
-    "yolov9e.pt",
-    "yolov9s.pt"
-]
+
+model_filenames = get_model_filenames(repo_id)
 
 def download_models(repo_id, model_id):
     # Download the selected model