Update app.py

改成智谱AI大模型，glm-4-9b

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer, AutoModel
 import fitz  # PyMuPDF for PDF handling
 
 # Function to extract text from PDF
@@ -21,7 +21,8 @@ def analyze_document(file, prompt):
         return "Unsupported file format. Please upload a PDF or TXT file."
     
     # Load model and tokenizer
-    model_name = "Alibaba-NLP/gte-Qwen1.5-7B-instruct"
+    # model_name = "Alibaba-NLP/gte-Qwen1.5-7B-instruct"
+    model_name = "THUDM/glm-4-9b"
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForCausalLM.from_pretrained(model_name)