brayden-gg commited on
Commit
da083b5
·
1 Parent(s): 9c37eb2

no more loading animation

Browse files
Files changed (1) hide show
  1. app.py +13 -12
app.py CHANGED
@@ -84,6 +84,7 @@ def update_writer_slider(val):
84
  def update_chosen_writers(writer1, writer2):
85
  net.clamp_mdn = 0
86
  chosen_writers[0], chosen_writers[1] = int(writer1.split(" ")[1]), int(writer2.split(" ")[1])
 
87
  all_loaded_data.clear()
88
  for writer_id in chosen_writers:
89
  loaded_data = dl.next_batch(TYPE='TRAIN', uid=writer_id, tids=list(range(num_samples)))
@@ -175,12 +176,12 @@ with gr.Blocks() as demo:
175
  writer_default_image = update_writer_slider(writer_weight)
176
  writer_output = gr.Image(writer_default_image)
177
 
178
- writer_submit.click(fn=update_writer_slider, inputs=[writer_slider], outputs=[writer_output])
179
- writer_slider.change(fn=update_writer_slider, inputs=[writer_slider], outputs=[writer_output])
180
- target_word.submit(fn=update_writer_word, inputs=[target_word], outputs=[writer_output])
181
 
182
- writer1.change(fn=update_chosen_writers, inputs=[writer1, writer2], outputs=[writer_slider, writer_output])
183
- writer2.change(fn=update_chosen_writers, inputs=[writer1, writer2], outputs=[writer_slider, writer_output])
184
 
185
  with gr.TabItem("Blend Characters"):
186
  with gr.Row():
@@ -194,10 +195,10 @@ with gr.Blocks() as demo:
194
  char_default_image = update_char_slider(char_weight)
195
  char_output = gr.Image(char_default_image)
196
 
197
- char_slider.change(fn=update_char_slider, inputs=[char_slider], outputs=[char_output])
198
 
199
- char1.change(fn=update_blend_chars, inputs=[char1, char2], outputs=[char_slider])
200
- char2.change(fn=update_blend_chars, inputs=[char1, char2], outputs=[char_slider])
201
 
202
  with gr.TabItem("Add Randomness"):
203
  mdn_word = gr.Textbox(label="Target Word", value=" ".join(mdn_words), max_lines=1)
@@ -212,9 +213,9 @@ with gr.Blocks() as demo:
212
  default_im = sample_mdn(net.scale_sd, net.clamp_mdn)
213
  mdn_output = gr.Image(default_im)
214
 
215
- max_rand.change(fn=sample_mdn, inputs=[scale_rand, max_rand], outputs=[mdn_output])
216
- scale_rand.change(fn=sample_mdn, inputs=[scale_rand, max_rand], outputs=[mdn_output])
217
- mdn_sample_button.click(fn=sample_mdn, inputs=[scale_rand, max_rand], outputs=[mdn_output])
218
- mdn_word.submit(fn=update_mdn_word, inputs=[mdn_word], outputs=[mdn_output])
219
 
220
  demo.launch()
 
84
  def update_chosen_writers(writer1, writer2):
85
  net.clamp_mdn = 0
86
  chosen_writers[0], chosen_writers[1] = int(writer1.split(" ")[1]), int(writer2.split(" ")[1])
87
+
88
  all_loaded_data.clear()
89
  for writer_id in chosen_writers:
90
  loaded_data = dl.next_batch(TYPE='TRAIN', uid=writer_id, tids=list(range(num_samples)))
 
176
  writer_default_image = update_writer_slider(writer_weight)
177
  writer_output = gr.Image(writer_default_image)
178
 
179
+ writer_submit.click(fn=update_writer_slider, inputs=[writer_slider], outputs=[writer_output], show_progress=False)
180
+ writer_slider.change(fn=update_writer_slider, inputs=[writer_slider], outputs=[writer_output], show_progress=False)
181
+ target_word.submit(fn=update_writer_word, inputs=[target_word], outputs=[writer_output], show_progress=False)
182
 
183
+ writer1.change(fn=update_chosen_writers, inputs=[writer1, writer2], outputs=[writer_slider, writer_output], show_progress=False)
184
+ writer2.change(fn=update_chosen_writers, inputs=[writer1, writer2], outputs=[writer_slider, writer_output], show_progress=False)
185
 
186
  with gr.TabItem("Blend Characters"):
187
  with gr.Row():
 
195
  char_default_image = update_char_slider(char_weight)
196
  char_output = gr.Image(char_default_image)
197
 
198
+ char_slider.change(fn=update_char_slider, inputs=[char_slider], outputs=[char_output], show_progress=False)
199
 
200
+ char1.change(fn=update_blend_chars, inputs=[char1, char2], outputs=[char_slider], show_progress=False)
201
+ char2.change(fn=update_blend_chars, inputs=[char1, char2], outputs=[char_slider], show_progress=False)
202
 
203
  with gr.TabItem("Add Randomness"):
204
  mdn_word = gr.Textbox(label="Target Word", value=" ".join(mdn_words), max_lines=1)
 
213
  default_im = sample_mdn(net.scale_sd, net.clamp_mdn)
214
  mdn_output = gr.Image(default_im)
215
 
216
+ max_rand.change(fn=sample_mdn, inputs=[scale_rand, max_rand], outputs=[mdn_output], show_progress=False)
217
+ scale_rand.change(fn=sample_mdn, inputs=[scale_rand, max_rand], outputs=[mdn_output], show_progress=False)
218
+ mdn_sample_button.click(fn=sample_mdn, inputs=[scale_rand, max_rand], outputs=[mdn_output], show_progress=False)
219
+ mdn_word.submit(fn=update_mdn_word, inputs=[mdn_word], outputs=[mdn_output], show_progress=False)
220
 
221
  demo.launch()