Update app.py

app.py

@@ -89,24 +89,32 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 final_answer = FinalAnswerTool()
 
-# 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'
+# Use the HF endpoint as suggested in the comment since there's an auth error with the direct model
 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
+    model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',  # Using the suggested endpoint
     custom_role_conversions=None,
 )
 
 # Import tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+try:
+    image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+except Exception as e:
+    print(f"Warning: Could not load image generation tool: {e}")
+    image_generation_tool = None
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
+# Create a list of tools, only adding ones that are successfully loaded
+tools = [final_answer, get_weather_forecast, get_current_time_in_timezone]
+if image_generation_tool is not None:
+    tools.append(image_generation_tool)
+
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_weather_forecast, get_current_time_in_timezone, image_generation_tool],  # Added all tools here
+    tools=tools,
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -116,4 +124,13 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-GradioUI(agent).launch()
+# Modified to handle potential errors in the Gradio UI
+try:
+    GradioUI(agent).launch()
+except TypeError as e:
+    if "unsupported operand type(s) for +=" in str(e):
+        print("Warning: There seems to be an issue with token counting in the Gradio UI.")
+        print("This could be due to the model not properly returning token count information.")
+        print("Consider modifying the Gradio_UI.py file to handle None values for token counts.")
+    else:
+        raise