Update app.py

app.py

@@ -25,9 +25,9 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model_name = "mistralai/Mistral-7B-v0.1"
-model = AutoModelForCausalLM.from_pretrained(model_name)
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+model_name = "microsoft/phi-2"
+model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 
 def generate_answer(question):
     inputs = tokenizer.encode("Question: " + question, return_tensors="pt")