dome272 commited on
Commit
a7a80b8
1 Parent(s): 85d3c26
Files changed (1) hide show
  1. app.py +6 -13
app.py CHANGED
@@ -10,8 +10,6 @@ from diffusers import WuerstchenDecoderPipeline, WuerstchenPriorPipeline
10
  from diffusers.pipelines.wuerstchen import DEFAULT_STAGE_C_TIMESTEPS
11
  from previewer.modules import Previewer
12
  from compel import Compel
13
- from share_btn import community_icon_html, loading_icon_html, share_js
14
-
15
  os.environ['TOKENIZERS_PARALLELISM'] = 'false'
16
 
17
 
@@ -103,7 +101,7 @@ def generate(
103
  for _ in range(len(DEFAULT_STAGE_C_TIMESTEPS)):
104
  r = next(prior_output)
105
  if isinstance(r, list):
106
- yield r, gr.update(visible=False)
107
  prior_output = r
108
 
109
  decoder_output = decoder_pipeline(
@@ -116,7 +114,7 @@ def generate(
116
  generator=generator,
117
  output_type="pil",
118
  ).images
119
- yield decoder_output, gr.update(visible=True)
120
 
121
 
122
  examples = [
@@ -142,10 +140,6 @@ with gr.Blocks(css="style.css") as demo:
142
  )
143
  run_button = gr.Button("Run", scale=0)
144
  result = gr.Gallery(label="Result", show_label=False)
145
- with gr.Group(elem_id="share-btn-container", visible=False) as share_group:
146
- community_icon = gr.HTML(community_icon_html)
147
- loading_icon = gr.HTML(loading_icon_html)
148
- share_button = gr.Button("Share to community", elem_id="share-btn")
149
  with gr.Accordion("Advanced options", open=False):
150
  negative_prompt = gr.Text(
151
  label="Negative prompt",
@@ -179,7 +173,7 @@ with gr.Blocks(css="style.css") as demo:
179
  num_images_per_prompt = gr.Slider(
180
  label="Number of Images",
181
  minimum=1,
182
- maximum=2,
183
  step=1,
184
  value=2,
185
  )
@@ -209,7 +203,7 @@ with gr.Blocks(css="style.css") as demo:
209
  decoder_num_inference_steps = gr.Slider(
210
  label="Decoder Inference Steps",
211
  minimum=10,
212
- maximum=12,
213
  step=1,
214
  value=12,
215
  )
@@ -245,7 +239,7 @@ with gr.Blocks(css="style.css") as demo:
245
  ).then(
246
  fn=generate,
247
  inputs=inputs,
248
- outputs=[result,share_group],
249
  api_name="run",
250
  )
251
  negative_prompt.submit(
@@ -272,7 +266,6 @@ with gr.Blocks(css="style.css") as demo:
272
  outputs=result,
273
  api_name=False,
274
  )
275
- share_button.click(None, [], [], _js=share_js)
276
-
277
  if __name__ == "__main__":
278
  demo.queue(max_size=20).launch()
 
10
  from diffusers.pipelines.wuerstchen import DEFAULT_STAGE_C_TIMESTEPS
11
  from previewer.modules import Previewer
12
  from compel import Compel
 
 
13
  os.environ['TOKENIZERS_PARALLELISM'] = 'false'
14
 
15
 
 
101
  for _ in range(len(DEFAULT_STAGE_C_TIMESTEPS)):
102
  r = next(prior_output)
103
  if isinstance(r, list):
104
+ yield r
105
  prior_output = r
106
 
107
  decoder_output = decoder_pipeline(
 
114
  generator=generator,
115
  output_type="pil",
116
  ).images
117
+ yield decoder_output
118
 
119
 
120
  examples = [
 
140
  )
141
  run_button = gr.Button("Run", scale=0)
142
  result = gr.Gallery(label="Result", show_label=False)
 
 
 
 
143
  with gr.Accordion("Advanced options", open=False):
144
  negative_prompt = gr.Text(
145
  label="Negative prompt",
 
173
  num_images_per_prompt = gr.Slider(
174
  label="Number of Images",
175
  minimum=1,
176
+ maximum=6,
177
  step=1,
178
  value=2,
179
  )
 
203
  decoder_num_inference_steps = gr.Slider(
204
  label="Decoder Inference Steps",
205
  minimum=10,
206
+ maximum=100,
207
  step=1,
208
  value=12,
209
  )
 
239
  ).then(
240
  fn=generate,
241
  inputs=inputs,
242
+ outputs=result,
243
  api_name="run",
244
  )
245
  negative_prompt.submit(
 
266
  outputs=result,
267
  api_name=False,
268
  )
269
+
 
270
  if __name__ == "__main__":
271
  demo.queue(max_size=20).launch()