Update app.py

app.py

@@ -6,6 +6,11 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 import json
 from Gradio_UI import GradioUI
+# Load model directly
+from transformers import AutoProcessor, AutoModel
+
+processor = AutoProcessor.from_pretrained("Qwen/Qwen2.5-Omni-3B")
+model = AutoModel.from_pretrained("Qwen/Qwen2.5-Omni-3B")
 
 duckduckgo_tool=DuckDuckGoSearchTool()
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
@@ -78,7 +83,7 @@ final_answer = FinalAnswerTool()
 model = HfApiModel(
 max_tokens=2096,
 temperature=0.5,
-model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' ,# it is possible that this model may be overloaded
+model_id= 'Qwen/Qwen2.5-Omni-3B',# it is possible that this model may be overloaded
 custom_role_conversions=None,
 )