Update app.py

app.py

@@ -70,15 +70,29 @@ with open("agent.json", "r") as f:
 model_config = agent_config["model"]["data"]
 
 # === BUILD MODEL ===
-model = LiteLLMModel(
-    model_id="gemini/gemini-2.0-flash-lite",
-    api_key=os.getenv("GEMINI_API_KEY"),
+# model = LiteLLMModel(
+#     model_id="gemini/gemini-2.0-flash-lite",
+#     api_key=os.getenv("GEMINI_API_KEY"),
+#     temperature=0.5,
+#     max_tokens=1024,
+# )
+
+# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
+# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' (!!!)
+
+model = HfApiModel(
+    model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+    token=os.getenv("HF_TOKEN"),
+    max_tokens=2096,
     temperature=0.5,
-    max_tokens=1024,
+    last_input_token_count=0,
+    last_output_token_count=0,
 )
 
 
 
+
+
 # === IMPORT TOOL FROM HUB ===
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)