Update app.py

app.py

@@ -3,6 +3,7 @@ import datetime
 import requests
 import pytz
 import yaml
+from huggingface_hub import login
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
@@ -53,8 +54,15 @@ custom_role_conversions=None,
 )
 
 
+# Carga el tool oficial de text-to-image
+image_generation_tool = load_tool(
+    "huggingface-tools/text-to-image",  # Nombre oficial
+    model="stabilityai/stable-diffusion-xl-base-1.0",  # Modelo específico
+    trust_remote_code=True
+)
+
 # Import tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+# image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)