Update app.py

app.py

@@ -4,8 +4,15 @@ import torch
 import re
 
 # 加载医学诊断模型
-model = AutoModelForSequenceClassification.from_pretrained("yxccai/ds-ai-model")
-tokenizer = AutoTokenizer.from_pretrained("yxccai/ds-ai-model")
+# 修改后（正确）
+from transformers import LlamaForSequenceClassification, LlamaTokenizer
+
+model = LlamaForSequenceClassification.from_pretrained(
+    "yxccai/ds-ai-model",
+    trust_remote_code=True  # 添加这行
+)
+tokenizer = LlamaTokenizer.from_pretrained("yxccai/ds-ai-model")
+
 
 # 疾病标签映射（必须与训练时完全一致！）
 disease_labels = [