Update app.py

app.py

@@ -15,7 +15,9 @@ else:
 # Load AI model (Optimized for T4 GPU)
 model_name = "TheBloke/Mistral-7B-Instruct-v0.1-AWQ"
 tokenizer = AutoTokenizer.from_pretrained(model_name, token=HF_TOKEN)
-model = AutoModelForCausalLM.from_pretrained(model_name, token=HF_TOKEN, device_map="auto")
+model = AutoModelForCausalLM.from_pretrained(
+    model_name, token=HF_TOKEN, device_map="auto", trust_remote_code=True
+)
 
 chatbot = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0)