RobotJelly commited on
Commit
2bd7f86
·
verified ·
1 Parent(s): a43593f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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("agents-course/text-to-image", trust_remote_code=True)
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()