Update app.py

app.py

@@ -5,7 +5,7 @@ from transformers import AutoTokenizer
 
 
 def get_available_models() -> list[str]:
-    """获取models目录下所有包含tokenizer.json的模型"""
+    """获取models目录下所有包含config.json的模型"""
     models_dir = "models"
     if not os.path.exists(models_dir):
         return []
@@ -13,9 +13,9 @@ def get_available_models() -> list[str]:
     available_models = []
     for model_name in os.listdir(models_dir):
         model_path = os.path.join(models_dir, model_name)
-        tokenizer_file = os.path.join(model_path, "tokenizer.json")
+        config_file = os.path.join(model_path, "config.json")
 
-        if os.path.isdir(model_path) and os.path.isfile(tokenizer_file):
+        if os.path.isdir(model_path) and os.path.isfile(config_file):
             available_models.append(model_name)
 
     return sorted(available_models)