Update app.py

app.py

@@ -46,7 +46,7 @@ CHAT_TEMPLATE="""{%- set default_system_message = "A user will ask you to solve
 
 if torch.cuda.is_available():
     model_id = "mistralai/Mistral-Small-24B-Instruct-2501"
-    model = AutoModelForCausalLM.from_pretrained("AlexHung29629/fix_magistral7", torch_dtype=torch.bfloat16, device_map="auto")
+    model = AutoModelForCausalLM.from_pretrained("AlexHung29629/fix_magistral_fc2", torch_dtype=torch.bfloat16, device_map="auto")
     tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-Small-24B-Instruct-2501")