Spaces:
Runtime error
Runtime error
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 |
|