Update app.py

app.py

@@ -83,14 +83,13 @@ with gr.Blocks(css=css) as demo:
     )
     with gr.Group():
         
-        with gr.Row(elem_id="prompt-container"):
-            with gr.Column():
-                gallery_output = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[4], height="auto")
+        with gr.Row():           
+            gallery_output = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[4], height="auto")
+        
         with gr.Row():
-            with gr.Column():
-                text_prompt = gr.Textbox(show_label=False, placeholder="Enter your prompt", max_lines=1)
-                negative_prompt = gr.Textbox(show_label=False, placeholder="Enter a negative", max_lines=1)
-                text_button = gr.Button("Generate")
+            text_prompt = gr.Textbox(show_label=False, placeholder="Enter your prompt", max_lines=1)
+            negative_prompt = gr.Textbox(show_label=False, placeholder="Enter a negative", max_lines=1)
+            text_button = gr.Button("Generate")