using only image generator tool & import from hub

app.py

@@ -44,20 +44,19 @@ model = HfApiModel(
 )
 
 # Import tool from Hub
-#image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-image_generation_tool = Tool.from_space(
-    "black-forest-labs/FLUX.1-schnell",
-    name="image_generator",
-    description="This tool creates an image according to a prompt, which is a text description.",
-    output_type = "image"    
-)
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+#image_generation_tool = Tool.from_space(
+#    "black-forest-labs/FLUX.1-schnell",
+#    name="image_generator",
+#    description="This tool creates an image according to a prompt, which is a text description."
+#)
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, image_generation_tool], ## add your tools here (don't remove final answer)
+    tools=[final_answer, image_generation_tool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,