Update app.py

app.py

@@ -3,8 +3,9 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 
 # Replace with your model name
-MODEL_NAME = "ssirikon/Gemma7b-bnb-Unsloth"
+#MODEL_NAME = "ssirikon/Gemma7b-bnb-Unsloth"
 #MODEL_NAME = "unsloth/gemma-7b-bnb-4bit"
+MODEL_NAME = "unsloth/mistral-7b-bnb-4bit"
 
 # Load the model and tokenizer
 model = AutoModelForCausalLM.from_pretrained(