Update app.py

app.py

@@ -37,8 +37,10 @@ with gr.Blocks(css=css) as demo:
             Direct Preference Optimization (DPO) for text-to-image diffusion models is a method to align diffusion models to text human preferences by directly optimizing on human comparison data.
         </p>
         """)
-        prompt_in = gr.Textbox(label="Prompt", value="An old man with a bird on his head")
-        submit_btn = gr.Button("Submit")
+        with gr.Group():
+            with gr.Column():
+                prompt_in = gr.Textbox(label="Prompt", value="An old man with a bird on his head")
+                submit_btn = gr.Button("Submit")
         result = gr.Image(label="DPO SDXL Result")
 
         gr.Examples(