Update app.py

app.py

@@ -12,8 +12,10 @@ model = AutoModelForCausalLM.from_pretrained(
     MODEL_NAME,
     torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
     device_map="auto",
+    trust_remote_code=True,
 )
 
+
 # Chat function
 def chat(user_input, history, max_new_tokens=200, temperature=1.0):
     messages = []