Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,13 @@ models = [
|
|
7 |
"nitrosocke/archer-diffusion",
|
8 |
"nitrosocke/elden-ring-diffusion",
|
9 |
"nitrosocke/spider-verse-diffusion",
|
10 |
-
"nitrosocke/modern-disney-diffusion"
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
]
|
12 |
|
13 |
prompt_prefixes = {
|
@@ -15,7 +21,13 @@ prompt_prefixes = {
|
|
15 |
models[1]: "archer style ",
|
16 |
models[2]: "elden ring style ",
|
17 |
models[3]: "spiderverse style ",
|
18 |
-
models[4]: "modern disney style "
|
|
|
|
|
|
|
|
|
|
|
|
|
19 |
}
|
20 |
|
21 |
current_model = models[0]
|
@@ -58,7 +70,8 @@ with gr.Blocks() as demo:
|
|
58 |
</h1>
|
59 |
</div>
|
60 |
<p style="margin-bottom: 10px; font-size: 94%">
|
61 |
-
Demo for multiple fine-tuned Stable Diffusion models, trained on different styles:
|
|
|
62 |
</p>
|
63 |
</div>
|
64 |
"""
|
@@ -67,7 +80,7 @@ with gr.Blocks() as demo:
|
|
67 |
|
68 |
with gr.Column():
|
69 |
model = gr.Dropdown(label="Model", choices=models, value=models[0])
|
70 |
-
prompt = gr.Textbox(label="Prompt", placeholder="
|
71 |
guidance = gr.Slider(label="Guidance scale", value=7.5, maximum=15)
|
72 |
steps = gr.Slider(label="Steps", value=50, maximum=100, minimum=2)
|
73 |
run = gr.Button(value="Run")
|
|
|
7 |
"nitrosocke/archer-diffusion",
|
8 |
"nitrosocke/elden-ring-diffusion",
|
9 |
"nitrosocke/spider-verse-diffusion",
|
10 |
+
"nitrosocke/modern-disney-diffusion",
|
11 |
+
"hakurei/waifu-diffusion",
|
12 |
+
"lambdalabs/sd-pokemon-diffusers",
|
13 |
+
"yuk/fuyuko-waifu-diffusion",
|
14 |
+
"AstraliteHeart/pony-diffusion",
|
15 |
+
"IfanSnek/JohnDiffusion",
|
16 |
+
"nousr/robo-diffusion"
|
17 |
]
|
18 |
|
19 |
prompt_prefixes = {
|
|
|
21 |
models[1]: "archer style ",
|
22 |
models[2]: "elden ring style ",
|
23 |
models[3]: "spiderverse style ",
|
24 |
+
models[4]: "modern disney style ",
|
25 |
+
models[5]: "",
|
26 |
+
models[6]: "",
|
27 |
+
models[7]: "",
|
28 |
+
models[8]: "",
|
29 |
+
models[9]: "",
|
30 |
+
models[10]: "",
|
31 |
}
|
32 |
|
33 |
current_model = models[0]
|
|
|
70 |
</h1>
|
71 |
</div>
|
72 |
<p style="margin-bottom: 10px; font-size: 94%">
|
73 |
+
Demo for multiple fine-tuned Stable Diffusion models, trained on different styles:
|
74 |
+
<a href="https://huggingface.co/nitrosocke/Arcane-Diffusion">Arcane</a>, <a href="https://huggingface.co/nitrosocke/archer-diffusion">Archer</a>, <a href="https://huggingface.co/nitrosocke/elden-ring-diffusion">Elden Ring</a>, <a href="https://huggingface.co/nitrosocke/spider-verse-diffusion">Spiderverse</a>, <a href="https://huggingface.co/nitrosocke/modern-disney-diffusion">Modern Disney</a>, <a href="https://huggingface.co/hakurei/waifu-diffusion">Waifu</a>, <a href="https://huggingface.co/lambdalabs/sd-pokemon-diffusers">Pokemon</a>, <a href="https://huggingface.co/yuk/fuyuko-waifu-diffusion">Fuyuko Waifu</a>, <a href="https://huggingface.co/AstraliteHeart/pony-diffusion">Pony</a>, <a href="https://huggingface.co/IfanSnek/JohnDiffusion">John</a>, <a href="https://huggingface.co/nousr/robo-diffusion">Robo</a>.
|
75 |
</p>
|
76 |
</div>
|
77 |
"""
|
|
|
80 |
|
81 |
with gr.Column():
|
82 |
model = gr.Dropdown(label="Model", choices=models, value=models[0])
|
83 |
+
prompt = gr.Textbox(label="Prompt", placeholder="Style prefix is applied automatically")
|
84 |
guidance = gr.Slider(label="Guidance scale", value=7.5, maximum=15)
|
85 |
steps = gr.Slider(label="Steps", value=50, maximum=100, minimum=2)
|
86 |
run = gr.Button(value="Run")
|