changed to mistral

app.py

@@ -1,11 +1,10 @@
 import gradio as gr
-from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-# Load the pre-trained model and tokenizer
-model_name = "google/flan-t5-base"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+# ── Choose the Mistral‑7B‑Instruct checkpoint ───────────────────────────────────
+model_name = "mistralai/Mistral-7B-Instruct-v0.1"
+model = AutoModelForCausalLM.from_pretrained(model_name)
 model = torch.compile(model)
 
 prompt_prefix = """