arranging ui better

app.py

@@ -175,21 +175,6 @@ with gr.Blocks(css=css) as demo:
                 step=1,
                 value=20,
             )
-            run_button.click(
-                fn=infer,
-                inputs=[
-                    prompt,
-                    negative_prompt,
-                    seed,
-                    randomize_seed,
-                    batch_size,
-                    width,
-                    height,
-                    guidance_scale,
-                    num_inference_steps,
-                ],
-                outputs=results,
-            )
 
     with gr.Column(elem_id="col-container"):
         with gr.Row():
@@ -204,6 +189,21 @@ with gr.Blocks(css=css) as demo:
                 )    
 
                 run_button = gr.Button("Run", scale=0)
+                run_button.click(
+                fn=infer,
+                inputs=[
+                    prompt,
+                    negative_prompt,
+                    seed,
+                    randomize_seed,
+                    batch_size,
+                    width,
+                    height,
+                    guidance_scale,
+                    num_inference_steps,
+                ],
+                outputs=results,
+                )
             results = []
                 
             with gr.Column():
@@ -211,6 +211,7 @@ with gr.Blocks(css=css) as demo:
                     with gr.Row():
                         for _ in range(2):
                             results.append(gr.Image(format="png", show_label=False, visible=True))
+
         
 
 demo.queue().launch()