AMfeta99 commited on
Commit
b0d6e04
·
verified ·
1 Parent(s): d7983da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
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
  # =========================================================