Update app.py

app.py

@@ -3,7 +3,6 @@ import datetime
 import requests
 import pytz
 import yaml
-import os
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
@@ -60,11 +59,11 @@ final_answer = FinalAnswerTool()
 model = HfApiModel(
 max_tokens=2096,
 temperature=0.5,
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
-token=os.getenv("HF_TOKEN"),
+model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
 custom_role_conversions=None,
 )
 
+model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-lite", api_key=os.getenv(key="AIzaSyB_QTuznTSXCMfY8l3YYfAn5sYyiHktVR8"))
 
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)