Update app.py

app.py

@@ -568,7 +568,7 @@ def create_ui():
                 input_image = gr.Image(
                     type="pil", 
                     label="Input Image", 
-                    height=400,
+                    # height=400,
                     show_download_button=True,
                     show_label=True,
                     scale=1,
@@ -634,7 +634,7 @@ def create_ui():
                 output_image = gr.Image(
                     label="Generated Image", 
                     type="pil", 
-                    height=400,
+                    # height=400,
                     interactive=False,
                     show_download_button=True,
                     scale=1,