app.py creación de imágenes

app.py

@@ -46,11 +46,7 @@ def conversor_unidades(arg1:str, arg2: int, arg3:str)-> str:
         return "Conversión no soportada entre esas unidades."
     except Exception as e:
         return f"Error en la conversión: {e}"
-
-
-
-    
-        
+     
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -89,7 +85,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[conversor_unidades, final_answer], ## add your tools here (don't remove final answer)
+    tools=[conversor_unidades, final_answer, image_generation_tool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,