Plat commited on
Commit
c360629
·
1 Parent(s): 29cfd71

chore: add info and examples

Browse files
Files changed (1) hide show
  1. app.py +43 -7
app.py CHANGED
@@ -93,7 +93,7 @@ def load_model(
93
  return model, label2id
94
 
95
 
96
- @spaces.GPU(duration=5)
97
  def generate_images(
98
  prompt: str,
99
  negative_prompt: str,
@@ -148,7 +148,7 @@ Pixel-space x-prediction flow-matching model for anime face generation, trained
148
  prompt = gr.TextArea(
149
  label="Prompt",
150
  info="Space-separated tags. Not all of danbooru tags are supported. See the link above for full supported tags.",
151
- value="general 1girl solo portrait looking_at_viewer blue_hair short_hair blush cat_ears open_mouth cat_ears animal_ears red_eyes white_background",
152
  placeholder="e.g.: general 1girl solo portrait looking_at_viewer",
153
  )
154
  negative_prompt = gr.TextArea(
@@ -161,8 +161,9 @@ Pixel-space x-prediction flow-matching model for anime face generation, trained
161
  minimum=1,
162
  maximum=100,
163
  value=25,
164
- step=4,
165
  label="Number of Steps",
 
166
  )
167
  cfg_scale = gr.Slider(
168
  minimum=1.0,
@@ -170,13 +171,15 @@ Pixel-space x-prediction flow-matching model for anime face generation, trained
170
  value=3.0,
171
  step=0.25,
172
  label="CFG Scale",
 
173
  )
174
  batch_size = gr.Slider(
175
  minimum=1,
176
  maximum=64,
177
- value=16,
178
  step=1,
179
  label="Batch Size",
 
180
  )
181
  size = gr.Slider(
182
  minimum=64,
@@ -184,6 +187,7 @@ Pixel-space x-prediction flow-matching model for anime face generation, trained
184
  value=256,
185
  step=64,
186
  label="Image Size",
 
187
  )
188
  seed = gr.Number(
189
  value=-1,
@@ -194,7 +198,7 @@ Pixel-space x-prediction flow-matching model for anime face generation, trained
194
  generate_button = gr.Button("Generate Images", variant="primary")
195
  output_gallery = gr.Gallery(
196
  label="Generated Images",
197
- columns=4,
198
  height="768px",
199
  preview=False,
200
  show_label=True,
@@ -203,9 +207,41 @@ Pixel-space x-prediction flow-matching model for anime face generation, trained
203
  gr.Examples(
204
  examples=[
205
  [
206
- "general 1girl solo portrait looking_at_viewer blue_hair short_hair blush cat_ears open_mouth cat_ears animal_ears red_eyes white_background",
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
207
  "retro_artstyle 1990s_(style) sketch",
208
- ]
209
  ],
210
  inputs=[prompt, negative_prompt],
211
  label="Examples",
 
93
  return model, label2id
94
 
95
 
96
+ @spaces.GPU(duration=10)
97
  def generate_images(
98
  prompt: str,
99
  negative_prompt: str,
 
148
  prompt = gr.TextArea(
149
  label="Prompt",
150
  info="Space-separated tags. Not all of danbooru tags are supported. See the link above for full supported tags.",
151
+ value="general 1girl solo portrait looking_at_viewer blue_hair short_hair blush open_mouth cat_ears animal_ears red_eyes white_background",
152
  placeholder="e.g.: general 1girl solo portrait looking_at_viewer",
153
  )
154
  negative_prompt = gr.TextArea(
 
161
  minimum=1,
162
  maximum=100,
163
  value=25,
164
+ step=1,
165
  label="Number of Steps",
166
+ info="Recommended: more than 20 steps for better quality.",
167
  )
168
  cfg_scale = gr.Slider(
169
  minimum=1.0,
 
171
  value=3.0,
172
  step=0.25,
173
  label="CFG Scale",
174
+ info="Recommended: 2.0 to 5.0 for balanced results.",
175
  )
176
  batch_size = gr.Slider(
177
  minimum=1,
178
  maximum=64,
179
+ value=25,
180
  step=1,
181
  label="Batch Size",
182
+ info="Number of images to generate in one batch.",
183
  )
184
  size = gr.Slider(
185
  minimum=64,
 
187
  value=256,
188
  step=64,
189
  label="Image Size",
190
+ info="Only 256x256 is supported in the current model. Other sizes may cause quality degradation.",
191
  )
192
  seed = gr.Number(
193
  value=-1,
 
198
  generate_button = gr.Button("Generate Images", variant="primary")
199
  output_gallery = gr.Gallery(
200
  label="Generated Images",
201
+ columns=5,
202
  height="768px",
203
  preview=False,
204
  show_label=True,
 
207
  gr.Examples(
208
  examples=[
209
  [
210
+ "general 1girl solo portrait looking_at_viewer blue_hair short_hair blush open_mouth cat_ears animal_ears red_eyes white_background",
211
+ "retro_artstyle 1990s_(style) sketch",
212
+ ],
213
+ [
214
+ "general 1girl aqua_eyes baseball_cap blonde_hair closed_mouth earrings green_background hat jewelry looking_at_viewer shirt short_hair simple_background solo portrait yellow_shirt",
215
+ "retro_artstyle 1990s_(style) sketch",
216
+ ],
217
+ [
218
+ "general 1girl solo portrait looking_at_viewer brown_hair ahoge long_hair :| expressionless closed_mouth swept_bangs pink_eyes pink_background simple_background dutch_angle",
219
+ "retro_artstyle 1990s_(style) sketch smile",
220
+ ],
221
+ [
222
+ "general 1girl solo portrait looking_at_viewer hatsune_miku twintails long_hair blue_eyes one_eye_closed simple_background green_background",
223
+ "retro_artstyle 1990s_(style) sketch",
224
+ ],
225
+ [
226
+ "general 1girl portrait looking_at_viewer sketch head_tilt white_background monochrome open_mouth long_hair",
227
+ "retro_artstyle 1990s_(style)",
228
+ ],
229
+ [
230
+ "general 1girl solo from_behind short_hair simple_background black_background",
231
+ "retro_artstyle 1990s_(style) sketch",
232
+ ],
233
+ [
234
+ "general 1girl portrait looking_to_the_side glasses",
235
+ "retro_artstyle 1990s_(style) sketch",
236
+ ],
237
+ [
238
+ "general 1girl portrait looking_at_viewer cat_ears purple_theme ;d forehead animal_ears animal_ear_fluff cat_ears",
239
+ "retro_artstyle 1990s_(style) sketch",
240
+ ],
241
+ [
242
+ "general 1girl portrait looking_at_viewer tongue sketch head_tilt white_background monochrome",
243
  "retro_artstyle 1990s_(style) sketch",
244
+ ],
245
  ],
246
  inputs=[prompt, negative_prompt],
247
  label="Examples",