UI changes

app.py

@@ -94,12 +94,12 @@ with gr.Blocks() as demo:
             ref_prompt = gr.Textbox(label="Reference prompt", value="Man laying on bed")
             
         with gr.Column():
-            
-            prompt1 = gr.Textbox(label="Prompt1", value="A man working on a laptop")
-            prompt2 = gr.Textbox(label="Prompt2", value="A man eating pizza")
-            prompt3 = gr.Textbox(label="Prompt3", value="A woman playing on saxophone")
-            run_button = gr.Button("Submit")
-        with gr.Column():
+            with gr.Group():
+                prompt1 = gr.Textbox(label="Prompt1", value="A man working on a laptop")
+                prompt2 = gr.Textbox(label="Prompt2", value="A man eating pizza")
+                prompt3 = gr.Textbox(label="Prompt3", value="A woman playing on saxophone")
+                run_button = gr.Button("Submit")
+        
             results = gr.Gallery()
     
     run_button.click(