anzorq commited on
Commit
6ba7689
1 Parent(s): 8131717

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -4
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: Arcane, Archer, Elden Ring, Spiderverse.
 
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="{} is added automatically".format(prompt_prefixes[current_model]))
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")