Update app.py

app.py

@@ -6,8 +6,7 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
-import os
-print(os.getenv('HF_TOKEN'))
+
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
@@ -65,7 +64,7 @@ print(prompt_templates)
     
 agent = CodeAgent(
     model=model,
-    tools=[get_current_time_in_timezone,final_answer], ## add your tools here (don't remove final answer)
+    tools=[image_generation_tool,final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,