Update app.py

app.py

@@ -44,19 +44,22 @@ def run_parallel_models(prompt):
     yield gr.update(), gr.update(), image
    
 #run_parallel_models.zerogpu = True
-
-with gr.Blocks() as demo:
+css = '''
+#gen_btn{height: 100%}
+#gen_column{align-self: stretch}
+'''
+with gr.Blocks(css=css) as demo:
     gr.Markdown("# Low Step Flux Comparison")
     gr.Markdown("Compare the quality (not the speed) of FLUX Schnell (4 steps), FLUX.1[dev] HyperFLUX (8 steps), FLUX.1[dev]-Turbo-Alpha (8 steps). It runs a bit slow as it's inferencing the three models.")
     with gr.Row():
         with gr.Column(scale=2):
             prompt = gr.Textbox(label="Prompt")
-        with gr.Column(scale=1, min_width=120):
-            submit = gr.Button("Run")
+        with gr.Column(scale=1, min_width=120, elem_id="gen_column"):
+            submit = gr.Button("Run", elem_id="gen_btn")
     with gr.Row():
-        schnell = gr.Image(label="FLUX Schnell (4 steps)")
         hyper = gr.Image(label="FLUX.1[dev] HyperFLUX (8 steps)")
         turbo = gr.Image(label="FLUX.1[dev]-Turbo-Alpha (8 steps)")
+        schnell = gr.Image(label="FLUX Schnell (4 steps)")
     
     gr.Examples(
         examples=[
@@ -76,6 +79,6 @@ with gr.Blocks() as demo:
         triggers=[submit.click, prompt.submit],
         fn=run_parallel_models,
         inputs=[prompt],
-        outputs=[schnell, hyper, turbo]
+        outputs=[hyper, turbo, schnell]
     )
 demo.launch()