Update app.py

app.py

@@ -152,39 +152,32 @@ h1 {
     background: #e64a19;
 }
 '''
-
 with gr.Blocks(css=css) as demo:
     gr.Markdown("# Image Generation with Face ID")
     gr.Markdown("Upload your face images and enter a prompt to generate images.")
 
-    with gr.Row():
-        with gr.Column():
-            images_input = gr.Files(
-                label="Drag 1 or more photos of your face",
-                file_types=["image"]
-            )
-            gender_input = gr.Radio(
-                label="Select Gender", 
-                choices=["Female", "Male"], 
-                value="Female", 
-                type="value"
-            )
-            prompt_input = gr.Textbox(
-                label="Enter your prompt",
-                placeholder="Describe the image you want to generate..."
-            )
-            run_button = gr.Button("Generate Image")
-
-        with gr.Column():
-            output_gallery = gr.Gallery(label="Generated Images")
-
-    # Define the event handler for the button click
+    images_input = gr.Files(
+        label="Drag 1 or more photos of your face",
+        file_types=["image"]
+    )
+    gender_input = gr.Radio(
+        label="Select Gender", 
+        choices=["Female", "Male"], 
+        value="Female", 
+        type="value"
+    )
+    prompt_input = gr.Textbox(
+        label="Enter your prompt",
+        placeholder="Describe the image you want to generate..."
+    )
+    run_button = gr.Button("Generate Image")
+    output_gallery = gr.Gallery(label="Generated Images")
+
     run_button.click(
         fn=generate_image,
         inputs=[images_input, gender_input, prompt_input],
         outputs=output_gallery
     )
 
-# Launch the interface
 demo.queue()
 demo.launch()