Update app.py

app.py

@@ -107,9 +107,14 @@ def generate_object_history(object_name):
 from smolagents.tools import Tool, load_tool
 image_generation_tool = load_tool("m-ric/text-to-image", cache=False, trust_remote_code=True)
 
+
 class WrappedTextToImageTool(Tool):
-    def run(self, *args, **kwargs):
-        return image_generation_tool(*args, **kwargs)
+    name = "text_to_image"
+    description = "Generates an image from a text prompt using the m-ric/text-to-image tool."
+    inputs = ["prompt"]  # Optional: only if smolagents uses this
+
+    def run(self, prompt: str) -> str:
+        return image_generation_tool(prompt)
 
 image_generation_tool = WrappedTextToImageTool()