Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -145,7 +145,9 @@ model = HfApiModel(
|
|
| 145 |
custom_role_conversions=None,
|
| 146 |
)
|
| 147 |
|
| 148 |
-
|
|
|
|
|
|
|
| 149 |
"black-forest-labs/FLUX.1-schnell",
|
| 150 |
name="image_generator", # You can name it whatever makes sense for your agent
|
| 151 |
description="Generate an image from a prompt"
|
|
|
|
| 145 |
custom_role_conversions=None,
|
| 146 |
)
|
| 147 |
|
| 148 |
+
from smolagents import Tool
|
| 149 |
+
|
| 150 |
+
image_generation_tool = Tool.from_space(
|
| 151 |
"black-forest-labs/FLUX.1-schnell",
|
| 152 |
name="image_generator", # You can name it whatever makes sense for your agent
|
| 153 |
description="Generate an image from a prompt"
|