test

app.py

@@ -63,7 +63,7 @@ This project is licensed under xxx.
 """
 
 NOTE = r"""
-🌟 🌟 🌟 Before clicking the generate button, **please wait until all images, prompts, and other components are fully loaded**, especially when using task examples. Otherwise, the inputs from the previous and current sessions may get mixed.
+❗❗❗ Before clicking the generate button, **please wait until all images, prompts, and other components are fully loaded**, especially when using task examples. Otherwise, the inputs from the previous and current sessions may get mixed.
 """
 
 def create_demo(model):
@@ -133,7 +133,7 @@ def create_demo(model):
                 )
                 
                 content_prompt = gr.Textbox(
-                    label="Content Description (Image caption, Editing instructions, etc.)",
+                    label="(Optional) Content Description (Image caption, Editing instructions, etc.)",
                     placeholder="Describe the content requirements...",
                     value="",
                     elem_id="content_prompt"
@@ -383,82 +383,114 @@ def create_demo(model):
         dense_prediction_tasks.click(
             partial(process_tasks, func=demo_tasks.process_dense_prediction_tasks), 
             inputs=[dense_prediction_tasks], 
-            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], show_progress=False, queue=False)
+            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], 
+            show_progress="full", 
+            queue=False)
 
         conditional_generation_tasks.click(
             partial(process_tasks, func=demo_tasks.process_conditional_generation_tasks), 
             inputs=[conditional_generation_tasks], 
-            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], show_progress=False, queue=False)
+            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], 
+            show_progress="full", 
+            queue=False)
 
         image_restoration_tasks.click(
             partial(process_tasks, func=demo_tasks.process_image_restoration_tasks), 
             inputs=[image_restoration_tasks], 
-            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], show_progress=False, queue=False)
+            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], 
+            show_progress="full", 
+            queue=False)
 
         style_transfer_tasks.click(
             partial(process_tasks, func=demo_tasks.process_style_transfer_tasks), 
             inputs=[style_transfer_tasks], 
-            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], show_progress=False, queue=False)
+            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], 
+            show_progress="full", 
+            queue=False)
 
         style_condition_fusion_tasks.click(
             partial(process_tasks, func=demo_tasks.process_style_condition_fusion_tasks), 
             inputs=[style_condition_fusion_tasks], 
-            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], show_progress=False, queue=False)
+            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], 
+            show_progress="full", 
+            queue=False)
             
         relighting_tasks.click(
             partial(process_tasks, func=demo_tasks.process_relighting_tasks), 
             inputs=[relighting_tasks], 
-            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], show_progress=False, queue=False)
+            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], 
+            show_progress="full", 
+            queue=False)
 
         tryon_tasks.click(
             partial(process_tasks, func=demo_tasks.process_tryon_tasks), 
             inputs=[tryon_tasks], 
-            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], show_progress=False, queue=False)
+            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], 
+            show_progress="full", 
+            queue=False)
         
         photodoodle_tasks.click(
             partial(process_tasks, func=demo_tasks.process_photodoodle_tasks), 
             inputs=[photodoodle_tasks], 
-            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], show_progress=False, queue=False)
+            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], 
+            show_progress="full", 
+            queue=False)
 
         editing_tasks.click(
             partial(process_tasks, func=demo_tasks.process_editing_tasks), 
             inputs=[editing_tasks], 
-            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], show_progress=False, queue=False)
+            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], 
+            show_progress="full", 
+            queue=False)
         
         unseen_tasks.click(
             partial(process_tasks, func=demo_tasks.process_unseen_tasks), 
             inputs=[unseen_tasks], 
-            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], show_progress=False, queue=False)
+            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], 
+            show_progress="full", 
+            queue=False)
 
         subject_driven_tasks.click(
             partial(process_tasks, func=demo_tasks.process_subject_driven_tasks), 
             inputs=[subject_driven_tasks], 
-            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], show_progress=False, queue=False)
+            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], 
+            show_progress="full", 
+            queue=False)
 
         style_transfer_with_subject_tasks.click(
             partial(process_tasks, func=demo_tasks.process_style_transfer_with_subject_tasks), 
             inputs=[style_transfer_with_subject_tasks], 
-            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], show_progress=False, queue=False)
+            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], 
+            show_progress="full", 
+            queue=False)
 
         condition_subject_fusion_tasks.click(
             partial(process_tasks, func=demo_tasks.process_condition_subject_fusion_tasks), 
             inputs=[condition_subject_fusion_tasks], 
-            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], show_progress=False, queue=False)
+            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], 
+            show_progress="full", 
+            queue=False)
 
         condition_subject_style_fusion_tasks.click(
             partial(process_tasks, func=demo_tasks.process_condition_subject_style_fusion_tasks), 
             inputs=[condition_subject_style_fusion_tasks], 
-            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], show_progress=False, queue=False)
+            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], 
+            show_progress="full", 
+            queue=False)
 
         editing_with_subject_tasks.click(
             partial(process_tasks, func=demo_tasks.process_editing_with_subject_tasks), 
             inputs=[editing_with_subject_tasks], 
-            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], show_progress=False, queue=False)
+            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], 
+            show_progress="full", 
+            queue=False)
 
         image_restoration_with_subject_tasks.click(
             partial(process_tasks, func=demo_tasks.process_image_restoration_with_subject_tasks), 
             inputs=[image_restoration_with_subject_tasks], 
-            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], show_progress=False, queue=False)
+            outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps], 
+            show_progress="full", 
+            queue=False)
         # Initialize grid
         model.set_grid_size(default_grid_h, default_grid_w)