Update app.py

app.py

@@ -7,11 +7,21 @@ import re
 # 修改后（正确）
 from transformers import LlamaForSequenceClassification, LlamaTokenizer
 
-model = LlamaForSequenceClassification.from_pretrained(
-    "yxccai/ds-ai-model",
-    trust_remote_code=True  # 添加这行
+# 1. 加载基础模型（示例使用Llama-7B）
+base_model = AutoModelForSequenceClassification.from_pretrained(
+    "unsloth/DeepSeek-R1-Distill-Llama-8B",
+    trust_remote_code=True
 )
-tokenizer = LlamaTokenizer.from_pretrained("yxccai/ds-ai-model")
+# 2. 加载你的适配器
+base_model.load_adapter("yxccai/ds-ai-app")  # 替换为你的仓库名
+
+# model = LlamaForSequenceClassification.from_pretrained(
+#     "yxccai/ds-ai-model",
+#     trust_remote_code=True  # 添加这行
+# )
+# tokenizer = LlamaTokenizer.from_pretrained("yxccai/ds-ai-model")
+
+tokenizer = AutoTokenizer.from_pretrained("yxccai/ds-ai-app")
 
 
 # 疾病标签映射（必须与训练时完全一致！）