Update app.py

app.py

@@ -329,9 +329,10 @@ Please draw boxes accordingly on the sketch pad.""".format(len(boxes), len(groun
         fix_seed, rand_seed, actual_mask, style_cond_image, clip_model=clip_model,
     )
     blank_samples = batch_size % 2 if batch_size > 1 else 0
-    gen_images = [gr.Image.update(value=x, visible=True) for i,x in enumerate(gen_images)] \
-                    + [gr.Image.update(value=None, visible=True) for _ in range(blank_samples)] \
-                    + [gr.Image.update(value=None, visible=False) for _ in range(4 - batch_size - blank_samples)]
+    gen_images = [gr.Image.update(value=x, visible=True) for i,x in enumerate(gen_images)] 
+                 # \
+                 #    + [gr.Image.update(value=None, visible=True) for _ in range(blank_samples)] \
+                 #    + [gr.Image.update(value=None, visible=False) for _ in range(4 - batch_size - blank_samples)]
 
     return gen_images + [state]
 
@@ -507,9 +508,10 @@ def clear(task, sketch_pad_trigger, batch_size, state,trigger_stage, switch_task
     sketch_pad_trigger = sketch_pad_trigger + 1
     trigger_stage = 0
     blank_samples = batch_size % 2 if batch_size > 1 else 0
-    out_images = [gr.Image.update(value=None, visible=True) for i in range(batch_size)] \
-                    + [gr.Image.update(value=None, visible=True) for _ in range(blank_samples)] \
-                    + [gr.Image.update(value=None, visible=False) for _ in range(4 - batch_size - blank_samples)]
+    out_images = [gr.Image.update(value=None, visible=True) for i in range(batch_size)] 
+    # \
+    #                 + [gr.Image.update(value=None, visible=True) for _ in range(blank_samples)] \
+    #                 + [gr.Image.update(value=None, visible=False) for _ in range(4 - batch_size - blank_samples)]
     state = {}
     return [None, sketch_pad_trigger, None, 1.0] + out_images + [state] + [trigger_stage]
 
@@ -676,7 +678,7 @@ with Blocks(
         clear_btn.click(
             clear,
             inputs=[task, sketch_pad_trigger, batch_size,trigger_stage, state],
-            outputs=[sketch_pad, sketch_pad_trigger, out_imagebox, image_scale, out_gen_1, out_gen_2, out_gen_3, out_gen_4, state, trigger_stage],
+            outputs=[sketch_pad, sketch_pad_trigger, out_imagebox, image_scale, out_gen_1, state, trigger_stage],
             queue=False)
     
         sketch_pad_trigger.change(
@@ -695,7 +697,7 @@ with Blocks(
                 append_grounding, style_cond_image,
                 state,
             ],
-            outputs=[out_gen_1, out_gen_2, out_gen_3, out_gen_4, state],
+            outputs=[out_gen_1, state],
             queue=True
         )
         init_white_trigger.change(