Update app.py

app.py

@@ -6,14 +6,16 @@ import re
 # 加载医学诊断模型
 # 修改后（正确）
 # from transformers import LlamaForSequenceClassification, LlamaTokenizer
-from transformers import AutoModelForSequenceClassification, AutoTokenizer, AutoConfig
+from transformers import LlamaForSequenceClassification, LlamaTokenizer, LlamaConfig
 import torch
 
-# ==== 绕过远程配置加载 ====
-# 强制从本地加载修正配置
-model = AutoModelForSequenceClassification.from_pretrained(
+# ==== 加载本地修正配置 ====
+config = LlamaConfig.from_pretrained("./fixed_config")  # 关键修改：强制本地配置
+
+# ==== 加载基础模型 ====
+model = LlamaForSequenceClassification.from_pretrained(
     "unsloth/DeepSeek-R1-Distill-Llama-8B",
-    config="./fixed_config/config.json",  # 关键修改：强制本地配置
+    config=config,  # 注入修正配置
     trust_remote_code=True
 )
 
@@ -21,7 +23,7 @@ model = AutoModelForSequenceClassification.from_pretrained(
 model.load_adapter("yxccai/ds-ai-app")
 
 # ==== 加载分词器 ===-
-tokenizer = AutoTokenizer.from_pretrained("yxccai/ds-ai-app")
+tokenizer = LlamaTokenizer.from_pretrained("yxccai/ds-ai-app")
 
 # 2. 加载你的适配器
 # model.load_adapter("yxccai/ds-ai-app")  # 替换为你的仓库名