Update app.py

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
-from transformers import Qwen2_5OmniModel, AutoProcessor
+from transformers import AutoModelForCausalLM, AutoProcessor
 import torch
 
 # Load model and processor
 model_name = "Qwen/Qwen2.5-Omni-3B"
-model = Qwen2_5OmniModel.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto")
+model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto")
 processor = AutoProcessor.from_pretrained(model_name)
 device = model.device
 
@@ -51,4 +51,4 @@ with gr.Blocks() as demo:
     )
 
 # Launch the app
-demo.launch()
+demo.launch()