lzyhha commited on
Commit
a801f0e
·
1 Parent(s): 4daee5e
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -384,7 +384,8 @@ def create_demo(model):
384
  partial(process_tasks, func=demo_tasks.process_dense_prediction_tasks),
385
  inputs=[dense_prediction_tasks],
386
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
387
- show_progress="full")
 
388
 
389
  conditional_generation_tasks.click(
390
  partial(process_tasks, func=demo_tasks.process_conditional_generation_tasks),
 
384
  partial(process_tasks, func=demo_tasks.process_dense_prediction_tasks),
385
  inputs=[dense_prediction_tasks],
386
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
387
+ show_progress="full",
388
+ show_progress_on=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps] + [generate_btn])
389
 
390
  conditional_generation_tasks.click(
391
  partial(process_tasks, func=demo_tasks.process_conditional_generation_tasks),