Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
 # Model setup
-model_name = "mistralai/Mistral-7B-Instruct-v0.3"
+model_name = "unsloth/Mistral-Small-3.1-24B-Instruct-2503"
 model = AutoModelForCausalLM.from_pretrained(
     model_name,
     torch_dtype=torch.bfloat16,