Update app.py

app.py

@@ -104,11 +104,12 @@ def build_prompt(chat_history, current_input_text, video_frames=None, image_file
 MODEL_ID = "mistralai/Mistral-Small-3.1-24B-Instruct-2503"
 SYSTEM_PROMPT = load_system_prompt(MODEL_ID, "SYSTEM_PROMPT.txt")
 
-tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
+tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(
     MODEL_ID,
     torch_dtype=torch.float16,
-    device_map="auto"
+    device_map="auto",
+    trust_remote_code=True
 ).to(device)
 model.eval()
 
@@ -201,4 +202,4 @@ demo = gr.ChatInterface(
 )
 
 if __name__ == "__main__":
-    demo.queue(max_size=20).launch(share=True)
+    demo.queue(max_size=20).launch(share=True)