Update app.py

app.py

@@ -169,10 +169,9 @@ with gr.Blocks(css=css, theme="pseudolab/huggingface-korea-theme") as dalle:
         """
     )
 
-    with gr.Row():
+    with gr.Row(elem_id="prompt-container"):
         image_output = gr.Image(type="pil", label="Output Image", elem_id="gallery")
     
-    with gr.Row(elem_id="prompt-container"):
         text_prompt = gr.Textbox(label="Prompt", placeholder="a cute cat", lines=1, elem_id="prompt-text-input")
         text_button = gr.Button("Generate", variant='primary', elem_id="gen-button")