AMfeta99 commited on
Commit
2345fb7
·
verified ·
1 Parent(s): 297cb09

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -107,9 +107,14 @@ def generate_object_history(object_name):
107
  from smolagents.tools import Tool, load_tool
108
  image_generation_tool = load_tool("m-ric/text-to-image", cache=False, trust_remote_code=True)
109
 
 
110
  class WrappedTextToImageTool(Tool):
111
- def run(self, *args, **kwargs):
112
- return image_generation_tool(*args, **kwargs)
 
 
 
 
113
 
114
  image_generation_tool = WrappedTextToImageTool()
115
 
 
107
  from smolagents.tools import Tool, load_tool
108
  image_generation_tool = load_tool("m-ric/text-to-image", cache=False, trust_remote_code=True)
109
 
110
+
111
  class WrappedTextToImageTool(Tool):
112
+ name = "text_to_image"
113
+ description = "Generates an image from a text prompt using the m-ric/text-to-image tool."
114
+ inputs = ["prompt"] # Optional: only if smolagents uses this
115
+
116
+ def run(self, prompt: str) -> str:
117
+ return image_generation_tool(prompt)
118
 
119
  image_generation_tool = WrappedTextToImageTool()
120