RobotJelly commited on
Commit
b1b01b1
·
verified ·
1 Parent(s): 730ac44

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)