Update app.py

app.py

@@ -104,8 +104,15 @@ def generate_object_history(object_name):
 
 #%% Initialization of tools and AI_Agent
 # Import text-to-image tool from Hub
+from smolagents.tools import Tool, load_tool
 image_generation_tool = load_tool("m-ric/text-to-image", cache=False)
 
+class WrappedTextToImageTool(Tool):
+    def run(self, *args, **kwargs):
+        return image_generation_tool(*args, **kwargs)
+
+image_generation_tool = WrappedTextToImageTool()
+
 # Import search tool from LangChain
 from transformers.agents.search import DuckDuckGoSearchTool
 search_tool = DuckDuckGoSearchTool()