Update app.py

app.py

@@ -5,7 +5,7 @@ from peft import PeftModel
 import os
 
 
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import AutoModel
 from transformers.generation.utils import GenerationConfig
 # 获取HF token（Spaces会自动提供）
 hf_token = os.environ.get("HF_TOKEN") or os.environ.get("HUGGING_FACE_HUB_TOKEN")
@@ -20,9 +20,8 @@ device = None
 
 try:
 
-    tokenizer = AutoTokenizer.from_pretrained("Go4miii/DISC-FinLLM", use_fast=False, trust_remote_code=True)
-    model = AutoModelForCausalLM.from_pretrained("Go4miii/DISC-FinLLM", device_map="auto", torch_dtype=torch.float16, trust_remote_code=True)
-    model.generation_config = GenerationConfig.from_pretrained("Go4miii/DISC-FinLLM")
+    # Load model directly
+    model = AutoModel.from_pretrained("Go4miii/DISC-FinLLM", trust_remote_code=True, dtype="auto")
 
 
 except Exception as e: