Update app.py

app.py

@@ -143,24 +143,26 @@ Swap faces using Flux.2 Klein 9B [Alissonerdx/BFS-Best-Face-Swap](https://huggin
         """)
 
         with gr.Row():
-            with gr.Column():
-                with gr.Row():
-                    reference_face = gr.Image(
-                        label="Reference Face",
-                        type="pil",
-                        sources=["upload"],
-                        elem_classes="image-container"
-                    )
-        
-                    target_image = gr.Image(
-                        label="Target Image (Body/Scene)",
-                        type="pil",
-                        sources=["upload"],
-                        elem_classes="image-container"
-                    )
-                run_button = gr.Button("Swap Face", variant="primary")
-                
-        
+            reference_face = gr.Image(
+                label="Reference Face",
+                type="pil",
+                sources=["upload"],
+                elem_classes="image-container"
+            )
+
+            target_image = gr.Image(
+                label="Target Image (Body/Scene)",
+                type="pil",
+                sources=["upload"],
+                elem_classes="image-container"
+            )
+
+            comparison_slider = gr.ImageSlider(
+                label="Before / After",
+                type="pil"
+            )
+
+        run_button = gr.Button("Swap Face", variant="primary", size="lg")
 
         with gr.Accordion("Advanced Settings", open=False):
             seed = gr.Slider(
@@ -210,12 +212,6 @@ Swap faces using Flux.2 Klein 9B [Alissonerdx/BFS-Best-Face-Swap](https://huggin
                 )
 
         seed_output = gr.Number(label="Seed Used", visible=False)
-            with gr.Column():
-                comparison_slider = gr.ImageSlider(
-                    label="Before / After",
-                    type="pil"
-                )
-
 
     # Auto-update dimensions when target image is uploaded
     target_image.upload(