Update app.py

app.py

@@ -185,110 +185,23 @@ if __name__ == "__main__":
     with gr.Blocks() as demo:
         
         with gr.Column():
-            # gr.HTML(title)
 
             with gr.Row():
                 with gr.Column():
-                    input_source = gr.Image(
-                        type="pil",
-                        label="Input Source"
-                    )
-                    input_target = gr.Image(
-                        type="pil",
-                        label="Input Target"
-                    )
+                    with gr.Row(equal_height=True):
+                        input_source = gr.Image(
+                            type="pil",
+                            label="Input Source"
+                        )
+                        input_target = gr.Image(
+                            type="pil",
+                            label="Input Target"
+                        )
                     run_button = gr.Button("Generate")
 
-                    # with gr.Row():
-                    #     with gr.Column(scale=2):
-                    #         prompt_input = gr.Textbox(label="Prompt (Optional)")
-                    #     with gr.Column(scale=1):
-                    #         run_button = gr.Button("Generate")
-
-                    # with gr.Row():
-                    #     target_ratio = gr.Radio(
-                    #         label="Expected Ratio",
-                    #         choices=["9:16", "16:9", "1:1", "Custom"],
-                    #         value="9:16",
-                    #         scale=2
-                    #     )
-                        
-                    #     alignment_dropdown = gr.Dropdown(
-                    #         choices=["Middle", "Left", "Right", "Top", "Bottom"],
-                    #         value="Middle",
-                    #         label="Alignment"
-                    #     )
-
-                    # with gr.Accordion(label="Advanced settings", open=False) as settings_panel:
-                    #     with gr.Column():
-                    #         with gr.Row():
-                    #             width_slider = gr.Slider(
-                    #                 label="Target Width",
-                    #                 minimum=720,
-                    #                 maximum=1536,
-                    #                 step=8,
-                    #                 value=720,  # Set a default value
-                    #             )
-                    #             height_slider = gr.Slider(
-                    #                 label="Target Height",
-                    #                 minimum=720,
-                    #                 maximum=1536,
-                    #                 step=8,
-                    #                 value=1280,  # Set a default value
-                    #             )
-                            
-                    #         num_inference_steps = gr.Slider(label="Steps", minimum=4, maximum=12, step=1, value=8)
-                    #         with gr.Group():
-                    #             overlap_percentage = gr.Slider(
-                    #                 label="Mask overlap (%)",
-                    #                 minimum=1,
-                    #                 maximum=50,
-                    #                 value=10,
-                    #                 step=1
-                    #             )
-                    #             with gr.Row():
-                    #                 overlap_top = gr.Checkbox(label="Overlap Top", value=True)
-                    #                 overlap_right = gr.Checkbox(label="Overlap Right", value=True)
-                    #             with gr.Row():
-                    #                 overlap_left = gr.Checkbox(label="Overlap Left", value=True)
-                    #                 overlap_bottom = gr.Checkbox(label="Overlap Bottom", value=True)
-                    #         with gr.Row():
-                    #             resize_option = gr.Radio(
-                    #                 label="Resize input image",
-                    #                 choices=["Full", "50%", "33%", "25%", "Custom"],
-                    #                 value="Full"
-                    #             )
-                    #             custom_resize_percentage = gr.Slider(
-                    #                 label="Custom resize (%)",
-                    #                 minimum=1,
-                    #                 maximum=100,
-                    #                 step=1,
-                    #                 value=50,
-                    #                 visible=False
-                    #             )
-                            
-                    #         with gr.Column():
-                    #             preview_button = gr.Button("Preview alignment and mask")
-                                
-                                
-                    # gr.Examples(
-                    #     examples=[
-                    #         ["./examples/example_1.webp", 1280, 720, "Middle"],
-                    #         ["./examples/example_2.jpg", 1440, 810, "Left"],
-                    #         ["./examples/example_3.jpg", 1024, 1024, "Top"],
-                    #         ["./examples/example_3.jpg", 1024, 1024, "Bottom"],
-                    #     ],
-                    #     inputs=[input_image, width_slider, height_slider, alignment_dropdown],
-                    # )
-
-                    
-
                 with gr.Column():
                     result = gr.Image(type='pil', label='Image Output')
-                    # use_as_input_button = gr.Button("Use as Input Image", visible=False)
-
-                    # history_gallery = gr.Gallery(label="History", columns=6, object_fit="contain", interactive=False)
-                    # preview_image = gr.Image(label="Preview")
+                    
         run_button.click(
             fn=infer_headswap,
             inputs=[input_source, input_target],