hardknee commited on
Commit
f8b79dd
·
verified ·
1 Parent(s): ec7c806

Update app.py

Browse files

Get rid of generate_image

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -63,8 +63,8 @@ def generate_image(description: str) -> str:
63
  description="Generate an image from a prompt"
64
  )
65
  # image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
66
- image = image_generation_tool.generate_image(description)
67
- return f"Image generated for prompt: {description}."
68
 
69
 
70
  # @tool
 
63
  description="Generate an image from a prompt"
64
  )
65
  # image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
66
+ image = image_generation_tool(description)
67
+ return f"Image generated for prompt: {description} is {image}."
68
 
69
 
70
  # @tool