Update app.py
Browse filesFixed the function for image generation
app.py
CHANGED
|
@@ -47,7 +47,7 @@ custom_role_conversions=None,
|
|
| 47 |
|
| 48 |
|
| 49 |
# Import tool from Hub
|
| 50 |
-
image_generation_tool =
|
| 51 |
"black-forest-labs/FLUX.1-schnell",
|
| 52 |
name="image_generator",
|
| 53 |
description="Generate an image from a prompt"
|
|
|
|
| 47 |
|
| 48 |
|
| 49 |
# Import tool from Hub
|
| 50 |
+
image_generation_tool = load_tool(
|
| 51 |
"black-forest-labs/FLUX.1-schnell",
|
| 52 |
name="image_generator",
|
| 53 |
description="Generate an image from a prompt"
|