Update app.py

app.py

@@ -48,15 +48,15 @@ with gr.Blocks() as demo:
       with gr.Column():
         with gr.Accordion(label='Enter upto 5 different initial prompts', open=True):
           with gr.Row(variant='panel'):
-            initial_prompt1 = gr.Textbox(value='', show_label=False, container=False, placeholder='a toy train')
-            initial_prompt2 = gr.Textbox(value='', show_label=False, container=False, placeholder='a toy airplane')
-            initial_prompt3 = gr.Textbox(value='', show_label=False, container=False, placeholder='a toy bicycle')
-            initial_prompt4 = gr.Textbox(value='', show_label=False, container=False, placeholder='a toy car')
-            initial_prompt5 = gr.Textbox(value='', show_label=False, container=False, placeholder='a toy boat')
+            initial_prompt1 = gr.Textbox(label='Initial prompt 1', value='', show_label=False, container=False, placeholder='a toy train')
+            initial_prompt2 = gr.Textbox(label='Initial prompt 2', value='', show_label=False, container=False, placeholder='a toy airplane')
+            initial_prompt3 = gr.Textbox(label='Initial prompt 3', value='', show_label=False, container=False, placeholder='a toy bicycle')
+            initial_prompt4 = gr.Textbox(label='Initial prompt 4', value='', show_label=False, container=False, placeholder='a toy car')
+            initial_prompt5 = gr.Textbox(label='Initial prompt 5', value='', show_label=False, container=False, placeholder='a toy boat')
         with gr.Row():
           style_prompt = gr.Textbox(label="Enter a style prompt", placeholder='macro photo, 3d game asset')
         btn = gr.Button("Generate a set of Style-aligned SDXL images",)
-    output = gr.Gallery(label="Style-Aligned SDXL", )
+    output = gr.Gallery(label="Style-Aligned SDXL Images", elem_id="gallery",columns=5, rows=1, object_fit="contain", height="auto",)
     
     btn.click(fn=style_aligned_sdxl, 
               inputs=[initial_prompt1, initial_prompt2, initial_prompt3, initial_prompt4, initial_prompt5, style_prompt],