Update app.py

app.py

@@ -6,7 +6,6 @@ import pytz
 import yaml
 import image
 from tools.final_answer import FinalAnswerTool
-from huggingface_hub import InferenceClient
 
 from Gradio_UI import GradioUI
 
@@ -36,15 +35,6 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-@tool
-def my_cutom_tool(prompt: str)-> image: #it's import to specify the return type
-    #Keep this format for the description / args / args description but feel free to modify the tool
-    """This tool creates an image according to a prompt, which is a text description.
-    Args:
-        prompt: The image generator prompt. Don't hesitate to add details in the prompt to make the image look better, like 'high-res, photorealistic', etc.
-    """
-    return InferenceClient("black-forest-labs/FLUX.1-schnell").text_to_image(prompt)
-
 final_answer = FinalAnswerTool()
 model = HfApiModel(
 max_tokens=2096,
@@ -53,6 +43,8 @@ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may
 custom_role_conversions=None,
 )
 
+# Import tool from Hub
+image_generator = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
@@ -69,5 +61,5 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-print(os.environ['HF_TOKEN'])
+print("\n".join([f"- {a}" for a in agent.toolbox.keys()]))
 GradioUI(agent).launch()