Update app.py

app.py

@@ -118,17 +118,18 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
     with main_interface:
         with gr.Row():
             with gr.Column(scale=1):
-                image_input_gallery = gr.Gallery(
-                    label="Upload one or more images here. Leave empty for text-to-image",
-                    file_types=["image"],
-                    height="auto"
-                )
-                        
-                prompt_input = gr.Textbox(
-                    label="Prompt",
-                    placeholder="Turns this photo into a masterpiece"
-                )
-                generate_button = gr.Button("Generate", variant="primary")
+                with gr.Group():
+                    image_input_gallery = gr.Gallery(
+                        label="Upload one or more images here. Leave empty for text-to-image",
+                        file_types=["image"],
+                        height="auto"
+                    )
+                            
+                    prompt_input = gr.Textbox(
+                        label="Prompt",
+                        placeholder="Turns this photo into a masterpiece"
+                    )
+                    generate_button = gr.Button("Generate", variant="primary")
 
             with gr.Column(scale=1):
                 output_image = gr.Image(label="Output", interactive=False, elem_id="output", type="filepath")