Update app.py

app.py

@@ -105,6 +105,13 @@ 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 = Tool.from_space(
+    space_id="m-ric/text-to-image",
+    name="text-to-image-generator",
+    description="Generate an image from a prompt"
+)
+'''
+
 image_generation_tool = load_tool("m-ric/text-to-image", cache=False, trust_remote_code=True)
 
 
@@ -117,7 +124,7 @@ class WrappedTextToImageTool(Tool):
         return image_generation_tool(prompt)
 
 image_generation_tool = WrappedTextToImageTool()
-
+'''
 # Import search tool from LangChain
 from transformers.agents.search import DuckDuckGoSearchTool
 search_tool = DuckDuckGoSearchTool()