Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ def download_models(model_id):
|
|
10 |
hf_hub_download("atalaydenknalbant/asl-models", filename=f"{model_id}", local_dir=f"./")
|
11 |
return f"./{model_id}"
|
12 |
|
13 |
-
def get_model_filenames("atalaydenknalbant/asl-models", file_extension=".pt"):
|
14 |
api = HfApi()
|
15 |
files = api.list_repo_files(repo_id)
|
16 |
model_filenames = [file for file in files if file.endswith(file_extension)]
|
|
|
10 |
hf_hub_download("atalaydenknalbant/asl-models", filename=f"{model_id}", local_dir=f"./")
|
11 |
return f"./{model_id}"
|
12 |
|
13 |
+
def get_model_filenames(repo_id="atalaydenknalbant/asl-models", file_extension=".pt"):
|
14 |
api = HfApi()
|
15 |
files = api.list_repo_files(repo_id)
|
16 |
model_filenames = [file for file in files if file.endswith(file_extension)]
|