Spaces:
Sleeping
Sleeping
specifying output type for image generation tool
Browse files
app.py
CHANGED
|
@@ -48,7 +48,8 @@ model = HfApiModel(
|
|
| 48 |
image_generation_tool = Tool.from_space(
|
| 49 |
"black-forest-labs/FLUX.1-schnell",
|
| 50 |
name="image_generator",
|
| 51 |
-
description="This tool creates an image according to a prompt, which is a text description."
|
|
|
|
| 52 |
)
|
| 53 |
|
| 54 |
with open("prompts.yaml", 'r') as stream:
|
|
|
|
| 48 |
image_generation_tool = Tool.from_space(
|
| 49 |
"black-forest-labs/FLUX.1-schnell",
|
| 50 |
name="image_generator",
|
| 51 |
+
description="This tool creates an image according to a prompt, which is a text description.",
|
| 52 |
+
output_type = "image"
|
| 53 |
)
|
| 54 |
|
| 55 |
with open("prompts.yaml", 'r') as stream:
|