patrickvonplaten multimodalart HF staff commited on
Commit
af85adf
1 Parent(s): 3868923

- Update UI (8caa43f5888e91e2d77ebec0eab2a48bd1c91bac)


Co-authored-by: Multimodal AI art <multimodalart@users.noreply.huggingface.co>

Files changed (2) hide show
  1. app.py +69 -51
  2. style.css +30 -18
app.py CHANGED
@@ -39,9 +39,9 @@ class Model:
39
 
40
 
41
  models = [
42
- Model("Stable Diffusion v1-4", "CompVis/stable-diffusion-v1-4"),
43
  # Model("Stable Diffusion v1-5", "runwayml/stable-diffusion-v1-5"),
44
- # Model("anything-v4.0", "andite/anything-v4.0"),
45
  ]
46
 
47
  MODELS = {m.name: m for m in models}
@@ -59,12 +59,12 @@ def error_str(error, title="Error"):
59
 
60
 
61
  def inference(
62
- model_name,
63
  prompt,
 
64
  guidance,
65
  steps,
66
- seed=0,
67
- neg_prompt="",
68
  ):
69
 
70
  print(psutil.virtual_memory()) # print memory usage
@@ -141,52 +141,61 @@ with gr.Blocks(css="style.css") as demo:
141
  gr.HTML(
142
  f"""
143
  <div class="finetuned-diffusion-div">
144
- <div>
145
- <h1>Stable Diffusion Latent Upscaler</h1>
 
 
 
146
  </div>
147
- <p>
148
- Demo for <a href="https://huggingface.co/stabilityai/sd-x2-latent-upscaler">Latent Diffusion Upscaling</a>
149
- </p>
150
- <p>
151
- Running on <b>{device}</b>
152
- </p>
153
- <p>You can also duplicate this space and upgrade to gpu by going to settings:<br>
154
  <a style="display:inline-block" href="https://huggingface.co/spaces/patrickvonplaten/finetuned_diffusion?duplicate=true"><img src="https://img.shields.io/badge/-Duplicate%20Space-blue?labelColor=white&style=flat&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAP5JREFUOE+lk7FqAkEURY+ltunEgFXS2sZGIbXfEPdLlnxJyDdYB62sbbUKpLbVNhyYFzbrrA74YJlh9r079973psed0cvUD4A+4HoCjsA85X0Dfn/RBLBgBDxnQPfAEJgBY+A9gALA4tcbamSzS4xq4FOQAJgCDwV2CPKV8tZAJcAjMMkUe1vX+U+SMhfAJEHasQIWmXNN3abzDwHUrgcRGmYcgKe0bxrblHEB4E/pndMazNpSZGcsZdBlYJcEL9Afo75molJyM2FxmPgmgPqlWNLGfwZGG6UiyEvLzHYDmoPkDDiNm9JR9uboiONcBXrpY1qmgs21x1QwyZcpvxt9NS09PlsPAAAAAElFTkSuQmCC&logoWidth=14" alt="Duplicate Space"></a></p>
 
155
  </div>
156
  """
157
  )
158
- with gr.Row():
159
 
160
- with gr.Column(scale=55):
161
- with gr.Group():
162
  model_name = gr.Dropdown(
163
  label="Model",
164
  choices=[m.name for m in models],
165
  value=models[0].name,
 
166
  )
167
 
168
- with gr.Row():
169
- prompt = gr.Textbox(
170
- label="Prompt",
171
- show_label=False,
172
- max_lines=2,
173
- placeholder="Enter prompt.",
174
- ).style(container=False)
175
- generate = gr.Button(value="Generate").style(
176
- rounded=(False, True, True, False)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
177
  )
178
-
179
- low_res_image = gr.Image(label="512px Image", shape=(512, 512))
180
- error_output = gr.Markdown()
181
-
182
- with gr.Column(scale=45):
183
- with gr.Tab("Options"):
184
  with gr.Group():
185
- neg_prompt = gr.Textbox(
186
- label="Negative prompt",
187
- placeholder="What to exclude from the image",
188
- )
189
-
190
  with gr.Row():
191
  guidance = gr.Slider(
192
  label="Guidance scale", value=7.5, maximum=15
@@ -202,36 +211,45 @@ with gr.Blocks(css="style.css") as demo:
202
  seed = gr.Slider(
203
  0, 2147483647, label="Seed (0 = random)", value=0, step=1
204
  )
 
205
 
206
- up_res_image = gr.Image(label="1024px Image", shape=(1024, 1024))
 
 
 
 
 
 
207
 
208
  inputs = [
209
- model_name,
210
  prompt,
 
211
  guidance,
212
  steps,
213
  seed,
214
- neg_prompt,
215
  ]
216
  outputs = [low_res_image, up_res_image, error_output]
217
  prompt.submit(inference, inputs=inputs, outputs=outputs)
218
  generate.click(inference, inputs=inputs, outputs=outputs)
219
 
220
- # ex = gr.Examples(
221
- # [
222
- # [models[0].name, "a photo of an astronaut high resolution, unreal engine, ultra realistic", 7.5, 50, 33, ""]
223
- # ],
224
- # inputs=[model_name, prompt, guidance, steps, seed, neg_prompt],
225
- # outputs=outputs,
226
- # fn=inference,
227
- # cache_examples=False,
228
- # )
229
-
 
 
230
  gr.HTML(
231
  """
232
  <div style="border-top: 1px solid #303030;">
233
  <br>
234
- <p>Models by 🤗 Hugging Face and others. ❤️</p>
235
  <p>This space uses the <a href="https://github.com/LuChengTHU/dpm-solver">DPM-Solver++</a> sampler by <a href="https://arxiv.org/abs/2206.00927">Cheng Lu, et al.</a>.</p>
236
  <p>This is a Demo Space For:<br>
237
  <a href="https://huggingface.co/stabilityai/sd-x2-latent-upscaler">Stability AI's Latent Upscaler</a>
 
39
 
40
 
41
  models = [
42
+ #Model("Stable Diffusion v1-4", "CompVis/stable-diffusion-v1-4"),
43
  # Model("Stable Diffusion v1-5", "runwayml/stable-diffusion-v1-5"),
44
+ Model("anything-v4.0", "andite/anything-v4.0"),
45
  ]
46
 
47
  MODELS = {m.name: m for m in models}
 
59
 
60
 
61
  def inference(
 
62
  prompt,
63
+ neg_prompt,
64
  guidance,
65
  steps,
66
+ seed,
67
+ model_name,
68
  ):
69
 
70
  print(psutil.virtual_memory()) # print memory usage
 
141
  gr.HTML(
142
  f"""
143
  <div class="finetuned-diffusion-div">
144
+ <div style="text-align: center">
145
+ <h1>Anything v4 model + <a href="https://huggingface.co/stabilityai/sd-x2-latent-upscaler">Stable Diffusion Latent Upscaler</a></h1>
146
+ <p>
147
+ Demo for the <a href="https://huggingface.co/andite/anything-v4.0">Anything v4</a> model hooked with the ultra-fast <a href="https://huggingface.co/stabilityai/sd-x2-latent-upscaler">Latent Upscaler</a>
148
+ </p>
149
  </div>
150
+ <!--
151
+ <p>To skip the queue, you can duplicate this Space<br>
 
 
 
 
 
152
  <a style="display:inline-block" href="https://huggingface.co/spaces/patrickvonplaten/finetuned_diffusion?duplicate=true"><img src="https://img.shields.io/badge/-Duplicate%20Space-blue?labelColor=white&style=flat&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAP5JREFUOE+lk7FqAkEURY+ltunEgFXS2sZGIbXfEPdLlnxJyDdYB62sbbUKpLbVNhyYFzbrrA74YJlh9r079973psed0cvUD4A+4HoCjsA85X0Dfn/RBLBgBDxnQPfAEJgBY+A9gALA4tcbamSzS4xq4FOQAJgCDwV2CPKV8tZAJcAjMMkUe1vX+U+SMhfAJEHasQIWmXNN3abzDwHUrgcRGmYcgKe0bxrblHEB4E/pndMazNpSZGcsZdBlYJcEL9Afo75molJyM2FxmPgmgPqlWNLGfwZGG6UiyEvLzHYDmoPkDDiNm9JR9uboiONcBXrpY1qmgs21x1QwyZcpvxt9NS09PlsPAAAAAElFTkSuQmCC&logoWidth=14" alt="Duplicate Space"></a></p>
153
+ -->
154
  </div>
155
  """
156
  )
 
157
 
158
+ with gr.Column(scale=100):
159
+ with gr.Group(visible=False):
160
  model_name = gr.Dropdown(
161
  label="Model",
162
  choices=[m.name for m in models],
163
  value=models[0].name,
164
+ visible=False
165
  )
166
 
167
+ with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
168
+ with gr.Column():
169
+ prompt = gr.Textbox(
170
+ label="Enter your prompt",
171
+ show_label=False,
172
+ max_lines=1,
173
+ placeholder="Enter your prompt",
174
+ elem_id="prompt-text-input",
175
+ ).style(
176
+ border=(True, False, True, True),
177
+ rounded=(True, False, False, True),
178
+ container=False,
179
+ )
180
+ neg_prompt = gr.Textbox(
181
+ label="Enter your negative prompt",
182
+ show_label=False,
183
+ max_lines=1,
184
+ placeholder="Enter a negative prompt",
185
+ elem_id="negative-prompt-text-input",
186
+ ).style(
187
+ border=(True, False, True, True),
188
+ rounded=(True, False, False, True),
189
+ container=False,
190
+ )
191
+ generate = gr.Button("Generate image").style(
192
+ margin=False,
193
+ rounded=(False, True, True, False),
194
+ full_width=False,
195
  )
196
+
197
+ with gr.Accordion("Advanced Options", open=False):
 
 
 
 
198
  with gr.Group():
 
 
 
 
 
199
  with gr.Row():
200
  guidance = gr.Slider(
201
  label="Guidance scale", value=7.5, maximum=15
 
211
  seed = gr.Slider(
212
  0, 2147483647, label="Seed (0 = random)", value=0, step=1
213
  )
214
+
215
 
216
+ with gr.Column(scale=100):
217
+ with gr.Row():
218
+ with gr.Column(scale=75):
219
+ up_res_image = gr.Image(label="Upscaled 1024px Image", shape=(1024, 1024))
220
+ with gr.Column(scale=25):
221
+ low_res_image = gr.Image(label="Original 512px Image", shape=(512, 512))
222
+ error_output = gr.Markdown()
223
 
224
  inputs = [
 
225
  prompt,
226
+ neg_prompt,
227
  guidance,
228
  steps,
229
  seed,
230
+ model_name,
231
  ]
232
  outputs = [low_res_image, up_res_image, error_output]
233
  prompt.submit(inference, inputs=inputs, outputs=outputs)
234
  generate.click(inference, inputs=inputs, outputs=outputs)
235
 
236
+ ex = gr.Examples(
237
+ [
238
+ ["a mecha robot in a favela", "low quality", 7.5, 25, 33, models[0].name],
239
+ ["the spirit of a tamagotchi wandering in the city of Paris", "low quality, bad render", 7.5, 50, 85, models[0].name],
240
+ ],
241
+ inputs=[prompt, neg_prompt, guidance, steps, seed, model_name],
242
+ outputs=outputs,
243
+ fn=inference,
244
+ cache_examples=True,
245
+ )
246
+ ex.dataset.headers = [""]
247
+
248
  gr.HTML(
249
  """
250
  <div style="border-top: 1px solid #303030;">
251
  <br>
252
+ <p>Space by 🤗 Hugging Face, models by Stability AI, andite, linaqruf and others ❤️</p>
253
  <p>This space uses the <a href="https://github.com/LuChengTHU/dpm-solver">DPM-Solver++</a> sampler by <a href="https://arxiv.org/abs/2206.00927">Cheng Lu, et al.</a>.</p>
254
  <p>This is a Demo Space For:<br>
255
  <a href="https://huggingface.co/stabilityai/sd-x2-latent-upscaler">Stability AI's Latent Upscaler</a>
style.css CHANGED
@@ -1,24 +1,36 @@
1
- .finetuned-diffusion-div div{
2
- display:inline-flex;
3
- align-items:center;
4
- gap:.8rem;
5
- font-size:1.75rem
6
  }
7
- .finetuned-diffusion-div div h1{
8
- font-weight:900;
9
- margin-bottom:7px
 
 
10
  }
11
- .finetuned-diffusion-div p{
12
- margin-bottom:10px;
13
- font-size:94%
 
14
  }
15
- a{
16
- text-decoration:underline
 
17
  }
18
- .tabs{
19
- margin-top:0;
20
- margin-bottom:0
 
21
  }
22
- #gallery{
23
- min-height:20rem
 
24
  }
 
 
 
 
 
 
 
 
 
 
1
+ .container {
2
+ max-width: 960px
 
 
 
3
  }
4
+
5
+ .finetuned-diffusion-div div {
6
+ align-items: center;
7
+ gap: .8rem;
8
+ font-size: 1.75rem
9
  }
10
+
11
+ .finetuned-diffusion-div div h1 {
12
+ font-weight: 900;
13
+ margin-bottom: 7px
14
  }
15
+
16
+ .finetuned-diffusion-div div p {
17
+ font-size: 50%
18
  }
19
+
20
+ .finetuned-diffusion-div p {
21
+ margin-bottom: 10px;
22
+ font-size: 94%
23
  }
24
+
25
+ a {
26
+ text-decoration: underline
27
  }
28
+
29
+ .tabs {
30
+ margin-top: 0;
31
+ margin-bottom: 0
32
+ }
33
+
34
+ #gallery {
35
+ min-height: 20rem
36
+ }