UI changes

app.py

@@ -96,9 +96,10 @@ with gr.Blocks(css=css) as demo:
         )
         with gr.Row():
             with gr.Column():
-                ref_path = gr.Image(type="filepath", value="./example_image/medieval-bed.jpeg")
-                ref_style = gr.Textbox(label="Reference style", value="medieval painting")
-                ref_prompt = gr.Textbox(label="Reference prompt", value="Man laying on bed")
+                with gr.Group():
+                    ref_path = gr.Image(type="filepath", value="./example_image/medieval-bed.jpeg")
+                    ref_style = gr.Textbox(label="Reference style", value="medieval painting")
+                    ref_prompt = gr.Textbox(label="Reference prompt", value="Man laying on bed")
                 
             with gr.Column():
                 with gr.Group():
@@ -107,7 +108,7 @@ with gr.Blocks(css=css) as demo:
                     prompt3 = gr.Textbox(label="Prompt3", value="A woman playing on saxophone")
                     run_button = gr.Button("Submit")
             
-                results = gr.Gallery()
+                results = gr.Gallery(columns=4)
     
     run_button.click(
         fn = run,