Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ def get_available_models():
|
|
13 |
available_models = []
|
14 |
for model_name in os.listdir(models_dir):
|
15 |
model_path = os.path.join(models_dir, model_name)
|
16 |
-
tokenizer_file = os.path.join(model_path, "
|
17 |
|
18 |
if os.path.isdir(model_path) and os.path.isfile(tokenizer_file):
|
19 |
available_models.append(model_name)
|
|
|
13 |
available_models = []
|
14 |
for model_name in os.listdir(models_dir):
|
15 |
model_path = os.path.join(models_dir, model_name)
|
16 |
+
tokenizer_file = os.path.join(model_path, "config.json")
|
17 |
|
18 |
if os.path.isdir(model_path) and os.path.isfile(tokenizer_file):
|
19 |
available_models.append(model_name)
|