Update app.py
Browse files
app.py
CHANGED
@@ -91,7 +91,7 @@ class TextToImageTool(Tool):
|
|
91 |
def forward(self, prompt):
|
92 |
return self.client.text_to_image(prompt)
|
93 |
'''
|
94 |
-
|
95 |
class TextToImageTool(Tool):
|
96 |
description = "This tool creates an image according to a prompt. Add details like 'high-res, photorealistic'."
|
97 |
name = "image_generator"
|
@@ -117,12 +117,10 @@ class TextToImageTool(Tool):
|
|
117 |
image = self.pipe(prompt).images[0]
|
118 |
return image
|
119 |
'''
|
120 |
-
|
121 |
image_generation_tool = Tool.from_space(
|
122 |
"black-forest-labs/FLUX.1-schnell",
|
123 |
name="image_generator",
|
124 |
-
description="Generate an image from a prompt"
|
125 |
-
)
|
126 |
|
127 |
|
128 |
# =========================================================
|
|
|
91 |
def forward(self, prompt):
|
92 |
return self.client.text_to_image(prompt)
|
93 |
'''
|
94 |
+
'''
|
95 |
class TextToImageTool(Tool):
|
96 |
description = "This tool creates an image according to a prompt. Add details like 'high-res, photorealistic'."
|
97 |
name = "image_generator"
|
|
|
117 |
image = self.pipe(prompt).images[0]
|
118 |
return image
|
119 |
'''
|
|
|
120 |
image_generation_tool = Tool.from_space(
|
121 |
"black-forest-labs/FLUX.1-schnell",
|
122 |
name="image_generator",
|
123 |
+
description="Generate an image from a prompt")
|
|
|
124 |
|
125 |
|
126 |
# =========================================================
|