Update app.py

app.py

@@ -79,8 +79,10 @@ with gr.Blocks() 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)")
     with gr.Row():
-        prompt = gr.Textbox(label="Prompt")
-        submit = gr.Button()
+        with gr.Column(scale=2):
+            prompt = gr.Textbox(label="Prompt")
+        with gr.Column(scale=1, min_width=120):
+            submit = gr.Button()
     with gr.Row():
         schnell = gr.Image(label="FLUX Schnell (4 steps)")
         hyper = gr.Image(label="FLUX.1[dev] HyperFLUX (8 steps)")