Update app.py

app.py

@@ -11,8 +11,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
 # Load the model and tokenizer
-tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen-7B")
-model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen-7B-Chat")
+tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen-7B-Chat", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen-7B-Chat", device_map="auto",trust_remote_code=True).eval()
 
 def generate_summary(prompt):
     # Add instructions to the prompt to signal that you want a summary