Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ def my_custom_tool(arg1:str, img: bool)-> str: #it's import to specify the retur
|
|
20 |
"""
|
21 |
try:
|
22 |
if img:
|
23 |
-
return Image.open(image_generation_tool(arg1))
|
24 |
else:
|
25 |
search_obj = DuckDuckGoSearchTool()
|
26 |
answer = search_obj(arg1)
|
|
|
20 |
"""
|
21 |
try:
|
22 |
if img:
|
23 |
+
return Image.open(image_generation_tool(arg1)).convert('RGB')
|
24 |
else:
|
25 |
search_obj = DuckDuckGoSearchTool()
|
26 |
answer = search_obj(arg1)
|