Update app.py

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
-from transformers import AutoModel
+from transformers import pipeline
 import spaces
 
 """
 For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
 """
-model = AutoModel.from_pretrained("Yoxas/autotrain-phi3-statistical", trust_remote_code=True)
+pipe = pipeline("text-generation", model="microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True)
 
 @spaces.GPU(duration=120)
 def respond(
@@ -28,7 +28,7 @@ def respond(
 
     response = ""
 
-    for message in model.text_generation(
+    for message in pipe.text_generation(
         messages,
         max_tokens=max_tokens,
         stream=True,