Update app.py

app.py

@@ -57,7 +57,7 @@ def generate(schedul):
 def ex(scheduler):
     t = time()
     print(ctime(t))
-    return generate(scheduler), generate(scheduler), generate(scheduler), generate(scheduler)
+    return generate(scheduler), generate(scheduler)
 
 demo = gr.Blocks(css="#img_size {max-height: 128px} .container {max-width: 730px; margin: auto;} .min-h-\[15rem\]{min-height: 5rem !important;}")
 
@@ -83,11 +83,8 @@ with demo:
     with gr.Row().style(equal_height=True):
         out = gr.Image(shape=(64,64), image_mode='RGBA', type='pil', elem_id='img_size')
         out2 = gr.Image(shape=(64,64), image_mode='RGBA', type='pil', elem_id='img_size')
-    with gr.Row().style(equal_height=True):
-        out3 = gr.Image(shape=(64,64), image_mode='RGBA', type='pil', elem_id='img_size')
-        out4 = gr.Image(shape=(64,64), image_mode='RGBA', type='pil', elem_id='img_size')
   greet_btn = gr.Button("Generate")
-  greet_btn.click(fn=ex, inputs=[model_name], outputs=[out, out2, out3, out4])
+  greet_btn.click(fn=ex, inputs=[model_name], outputs=[out, out2)
   gr.HTML(
             """
                 <div class="footer">