Orion-zhen commited on
Commit
d0dc6a4
·
verified ·
1 Parent(s): 41f8d0a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, "tokenizer.json")
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)