Update app.py

app.py

@@ -6,21 +6,24 @@ import re
 # 加载医学诊断模型
 # 修改后（正确）
 # from transformers import LlamaForSequenceClassification, LlamaTokenizer
-from transformers import AutoModelForSequenceClassification, AutoConfig
+from transformers import AutoConfig, AutoModelForSequenceClassification
 
-# 1. 加载原始配置
+# 1. 加载配置并强制修正
 config = AutoConfig.from_pretrained("unsloth/DeepSeek-R1-Distill-Llama-8B")
 
-# 2. 修正RoPE参数
+# 完全覆盖RoPE参数（关键步骤）
 config.rope_scaling = {
     "type": "linear",  # 必须字段
-    "factor": config.rope_scaling["factor"]  # 保留原缩放因子
+    "factor": 8.0      # 从原配置中提取的值
 }
 
-# 3. 加载模型（使用修正后的配置）
+# 2. 保存修正后的配置到本地
+config.save_pretrained("./fixed_config")
+
+# 3. 从本地加载修正后的配置
 model = AutoModelForSequenceClassification.from_pretrained(
     "unsloth/DeepSeek-R1-Distill-Llama-8B",
-    config=config,
+    config="./fixed_config",  # 指定本地修正配置
     trust_remote_code=True
 )