Spaces:
Sleeping
Sleeping
Update app.py
Browse filestry adding save to picture
app.py
CHANGED
@@ -64,6 +64,7 @@ def generate_image(description: str) -> str:
|
|
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 |
|
|
|
64 |
)
|
65 |
# image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
|
66 |
image = image_generation_tool(description)
|
67 |
+
image.save("my picture.png")
|
68 |
return f"Image generated for prompt: {description} is {image}."
|
69 |
|
70 |
|