Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ def my_custom_tool(arg1:str, img: bool)-> Any: #it's import to specify the retur
|
|
22 |
try:
|
23 |
if img:
|
24 |
# Import tool from Hub
|
25 |
-
image_generation_tool = load_tool("
|
26 |
image_generation_tool(arg1)
|
27 |
else:
|
28 |
search_obj = DuckDuckGoSearchTool()
|
|
|
22 |
try:
|
23 |
if img:
|
24 |
# Import tool from Hub
|
25 |
+
image_generation_tool = load_tool("dwb2023/text-to-image", trust_remote_code=True)
|
26 |
image_generation_tool(arg1)
|
27 |
else:
|
28 |
search_obj = DuckDuckGoSearchTool()
|