Upload app.py

app.py

@@ -183,7 +183,7 @@ def clear(task, sketch_pad_trigger, batch_size, state, switch_task=False):
     if task != 'Grounded Inpainting':
         sketch_pad_trigger = sketch_pad_trigger + 1
     blank_samples = batch_size % 2 if batch_size > 1 else 0
-    out_images = [gr.Image.change(value=None, visible=True) for i in range(batch_size)]
+    out_images = [None]
     # state = {}
     return [None, sketch_pad_trigger, None, 1.0] + out_images + [{}]
 
@@ -296,9 +296,9 @@ def main():
                                guidance_scale)
 
         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.change(fn=None, show_api=True) for _ in range(blank_samples)] \
-                     + [gr.Image.change(fn=None, show_api=False) for _ in range(4 - batch_size - blank_samples)]
+        gen_images = [x for i, x in enumerate(gen_images)] \
+                     + [_ for _ in range(blank_samples)] \
+                     + [_ for _ in range(4 - batch_size - blank_samples)]
 
         return gen_images + [state]