Update app.py

app.py

@@ -185,7 +185,7 @@ with gr.Blocks(css=css) as demo:
     )
     with gr.Group():
         with gr.Box():
-            with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
+            with gr.Row().style(mobile_collapse=False, equal_height=True):
                 with gr.Column():
                     gallery_output = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
             with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):