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")
|