prova

app.py

@@ -52,24 +52,14 @@ custom_role_conversions=None,
 )
 
 
-@tool
-def image_generation_tool(prompt:str, token:str) -> str:
-    """A tool that generates an image from a text prompt.
-
-    Args:
-        prompt: The text prompt to generate an image from.
-        token: The API token to use for the image generation tool.
-    """
-    # Call the image generation tool with the provided prompt and API token
-    tool = load_tool("agents-course/text-to-image", trust_remote_code=True, token=token)
-    return tool(prompt=prompt)
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True, token=token)
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, python_interpreter, 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,