Update app.py

app.py

@@ -143,74 +143,78 @@ Swap faces using Flux.2 Klein 9B [Alissonerdx/BFS-Best-Face-Swap](https://huggin
         """)
 
         with gr.Row():
-            reference_face = gr.Image(
-                label="Reference Face",
-                type="pil",
-                sources=["upload"],
-                elem_classes="image-container"
-            )
+            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", visible=False)
+                with gr.Accordion("Advanced Settings", open=False):
+                    seed = gr.Slider(
+                        label="Seed",
+                        minimum=0,
+                        maximum=MAX_SEED,
+                        step=1,
+                        value=0,
+                    )
+        
+                    randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
+        
+                    with gr.Row():
+                        width = gr.Slider(
+                            label="Width",
+                            minimum=256,
+                            maximum=1024,
+                            step=8,
+                            value=1024,
+                        )
+        
+                        height = gr.Slider(
+                            label="Height",
+                            minimum=256,
+                            maximum=1024,
+                            step=8,
+                            value=1024,
+                        )
+        
+                    with gr.Row():
+                        num_inference_steps = gr.Slider(
+                            label="Inference Steps",
+                            minimum=1,
+                            maximum=20,
+                            step=1,
+                            value=4,
+                            info="Number of denoising steps (4 is optimal for distilled model)"
+                        )
+        
+                        guidance_scale = gr.Slider(
+                            label="Guidance Scale",
+                            minimum=0.0,
+                            maximum=5.0,
+                            step=0.1,
+                            value=1.0,
+                            info="How closely to follow the prompt (1.0 is optimal for distilled model)"
+                        )
 
-            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", visible=False)
-
-        with gr.Accordion("Advanced Settings", open=False):
-            seed = gr.Slider(
-                label="Seed",
-                minimum=0,
-                maximum=MAX_SEED,
-                step=1,
-                value=0,
-            )
-
-            randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
-
-            with gr.Row():
-                width = gr.Slider(
-                    label="Width",
-                    minimum=256,
-                    maximum=1024,
-                    step=8,
-                    value=1024,
-                )
-
-                height = gr.Slider(
-                    label="Height",
-                    minimum=256,
-                    maximum=1024,
-                    step=8,
-                    value=1024,
-                )
-
-            with gr.Row():
-                num_inference_steps = gr.Slider(
-                    label="Inference Steps",
-                    minimum=1,
-                    maximum=20,
-                    step=1,
-                    value=4,
-                    info="Number of denoising steps (4 is optimal for distilled model)"
-                )
-
-                guidance_scale = gr.Slider(
-                    label="Guidance Scale",
-                    minimum=0.0,
-                    maximum=5.0,
-                    step=0.1,
-                    value=1.0,
-                    info="How closely to follow the prompt (1.0 is optimal for distilled model)"
-                )
+        
 
+        
         seed_output = gr.Number(label="Seed Used", visible=False)
 
     # Auto-update dimensions when target image is uploaded