Update app.py

app.py

@@ -5,6 +5,7 @@ from pathlib import Path
 import re
 from Model import OmniPathWithInterTaskAttention
 from transformers import AutoModelForCausalLM, AutoTokenizer
+import transformers
 import os
 
 # 强制设置 Gradio 为英文环境
@@ -48,7 +49,7 @@ def load_models():
     
     # 2. Load text generation model
     # llm_model_name = "Qwen/Qwen3-0.6B"
-    llm_model_name = "meta-llama/Meta-Llama-3.1-8B-Instruct"
+    llm_model_name = "meta-llama/Meta-Llama-3-8B"
     tokenizer = AutoTokenizer.from_pretrained(llm_model_name)
     llm_model = AutoModelForCausalLM.from_pretrained(
         llm_model_name,