Update app.py

app.py

@@ -1,5 +1,9 @@
 import gradio as gr
-from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
+
+# Load model directly
+from transformers import AutoModel
+model = AutoModel.from_pretrained("fixie-ai/ultravox-v0_5-llama-3_2-1b", trust_remote_code=True)
+
 
 model_id = "fixie-ai/ultravox-v0_5-llama-3_2-1b"