Update app.py

app.py

@@ -169,11 +169,11 @@ with gr.Blocks(css=css, theme="pseudolab/huggingface-korea-theme") as dalle:
         """
     )
 
-    with gr.Row(elem_id="prompt-container"):
+    with gr.Row():
         image_output = gr.Image(type="pil", label="Output Image", elem_id="gallery")
-    
-        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")
+        with gr.Column(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")
 
     with gr.Accordion("Advanced settings", open=False):
         negative_prompt = gr.Textbox(label="Negative Prompt", value="text, blurry, fuzziness", lines=1, elem_id="negative-prompt-text-input")