Update app.py

app.py

@@ -1,8 +1,11 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 
-# Load the model and tokenizer from Hugging Face's model hub
-model = AutoModelForCausalLM.from_pretrained("microsoft/Phi-3-mini-4k-instruct")
+# Load the model and tokenizer from Hugging Face's model hub with trust_remote_code=True
+model = AutoModelForCausalLM.from_pretrained(
+    "microsoft/Phi-3-mini-4k-instruct", 
+    trust_remote_code=True  # Allow custom code execution
+)
 tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-4k-instruct")
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
@@ -25,3 +28,4 @@ iface = gr.Interface(
 
 if __name__ == "__main__":
     iface.launch()
+