Update app.py

app.py

@@ -54,14 +54,6 @@ with gr.Blocks() as demo:
         stop_button2 = gr.Button('Stop', variant = 'secondary', interactive = False)
         gen_button2.click(lambda s: gr.update(interactive = True), None, stop_button2)
         
-        with gr.Row():
-            output2 = [gr.Image(label = '', min_width=480) for _ in range(max_images)]
-
-        for i, o in enumerate(output2):
-            img_i = gr.Number(i, visible = False)
-            num_images.change(lambda i, n: gr.update(visible = (i < n)), [img_i, num_images], o, show_progress = False)
-            gen_event2 = gen_button2.click(lambda i, n, m, t: gen_fn(m, t) if (i < n) else None, [img_i, num_images, model_choice2, txt_input2], o)
-            stop_button2.click(lambda s: gr.update(interactive = False), None, stop_button2, cancels = [gen_event2])
         gr.HTML(
         """
             <div style="text-align: center; max-width: 1200px; margin: 0 auto;">
@@ -74,6 +66,15 @@ with gr.Blocks() as demo:
             </div>
         """
                )
+        
+        with gr.Row():
+            output2 = [gr.Image(label = '') for _ in range(max_images)]
+
+        for i, o in enumerate(output2):
+            img_i = gr.Number(i, visible = False)
+            num_images.change(lambda i, n: gr.update(visible = (i < n), min_width=480), [img_i, num_images], o, show_progress = False)
+            gen_event2 = gen_button2.click(lambda i, n, m, t: gen_fn(m, t) if (i < n) else None, [img_i, num_images, model_choice2, txt_input2], o)
+            stop_button2.click(lambda s: gr.update(interactive = False), None, stop_button2, cancels = [gen_event2])
         with gr.Row():
             gr.HTML(
     """