Update app.py

app.py

@@ -10,7 +10,14 @@ def download_models(model_id):
     hf_hub_download("atalaydenknalbant/asl-models", filename=f"{model_id}", local_dir=f"./")
     return f"./{model_id}"
 
+def get_model_filenames("atalaydenknalbant/asl-models", 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(repo_id)
+print("Model filenames:", model_filenames)
 
 box_annotator = sv.BoxAnnotator()
 category_dict = {0: 'A', 1: 'B', 2: 'C', 3: 'D', 4: 'E', 5: 'F', 6: 'G', 7: 'H', 8: 'I',
@@ -44,16 +51,21 @@ def app():
             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=[
-                        "yolov10x.pt",
-                        "yolov10s.pt",
-                        "yolov9e.pt",
-                        "yolov9s.pt",
-                        "yolov8x.pt",
-                    ],
-                    value="yolov10s.pt",
+                    choices=model_filenames,
+                    value=model_filenames[0] if model_filenames else "",
                 )
                 conf_threshold = gr.Slider(
                     label="Confidence Threshold",