Update app.py

app.py

@@ -105,6 +105,8 @@ 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="image_generator",
@@ -112,9 +114,15 @@ image_generation_tool = Tool.from_space(
 )
 '''
 
-image_generation_tool = load_tool("m-ric/text-to-image", cache=False, trust_remote_code=True)
+from smolagents import load_tool
 
+image_generation_tool = load_tool(
+    repo_id="m-ric/text-to-image",  # The Hugging Face Space repo
+    trust_remote_code=True,         # ✅ Required to execute downloaded code
+    cache=False                     # Optional: disable cache for latest version
+)
 
+'''
 class WrappedTextToImageTool(Tool):
     name = "text_to_image"
     description = "Generates an image from a text prompt using the m-ric/text-to-image tool."
@@ -125,6 +133,8 @@ class WrappedTextToImageTool(Tool):
 
 image_generation_tool = WrappedTextToImageTool()
 '''
+
+
 # Import search tool from LangChain
 #from transformers.agents.search import DuckDuckGoSearchTool
 search_tool = DuckDuckGoSearchTool()