Spaces:
Running
on
Zero
Running
on
Zero
Plat
commited on
Commit
·
c360629
1
Parent(s):
29cfd71
chore: add info and examples
Browse files
app.py
CHANGED
|
@@ -93,7 +93,7 @@ def load_model(
|
|
| 93 |
return model, label2id
|
| 94 |
|
| 95 |
|
| 96 |
-
@spaces.GPU(duration=
|
| 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
|
| 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=
|
| 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=
|
| 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=
|
| 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
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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",
|