Update app.py

app.py

@@ -1,4 +1,10 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
+from smolagents import (
+    CodeAgent,
+    DuckDuckGoSearchTool,
+    HfApiModel,
+    load_tool,
+    tool
+)
 import datetime
 import pytz
 import yaml
@@ -50,15 +56,15 @@ model = HfApiModel(
 )
 
 # —————————————————————————————————————————————
-# 4) Load & instantiate external tools
+# 4) Load external tools
+#    * **Do not** call the returned object!
 # —————————————————————————————————————————————
 
-# DuckDuckGo search must be instantiated, not passed as a class
 duck_search = DuckDuckGoSearchTool()
-
-# Text-to-image tool from the Hub often returns a class—instantiate it here
-ImageGenClass = load_tool("agents-course/text-to-image", trust_remote_code=True)
-image_generation_tool = ImageGenClass()
+image_generation_tool = load_tool(
+    "agents-course/text-to-image",
+    trust_remote_code=True
+)
 
 # —————————————————————————————————————————————
 # 5) Load your custom prompt templates
@@ -74,11 +80,11 @@ with open("prompts.yaml", "r") as stream:
 agent = CodeAgent(
     model=model,
     tools=[
-        final_answer,                    # must be an instance
-        my_custom_tool,                  # your custom @tool
-        get_current_time_in_timezone,    # your custom @tool
-        duck_search,                     # instantiated
-        image_generation_tool,           # instantiated
+        final_answer,                 # instance of FinalAnswerTool
+        my_custom_tool,               # your @tool
+        get_current_time_in_timezone, # your @tool
+        duck_search,                  # instance of DuckDuckGoSearchTool
+        image_generation_tool,        # the load_tool() instance
     ],
     max_steps=6,
     verbosity_level=1,