Update app.py

app.py

@@ -153,7 +153,7 @@ def generate_image(prompt,
         np.random.seed(seed)
     width = 384
     height = 384
-    parallel_size = 5
+    parallel_size = 4
     
     with torch.no_grad():
         messages = [{'role': '<|User|>', 'content': prompt},
@@ -239,7 +239,7 @@ with gr.Blocks(css=css) as demo:
     
         generation_button = gr.Button("Generate Images")
     
-        image_output = gr.Gallery(label="Generated Images", columns=2, rows=2)
+        image_output = gr.Gallery(label="Generated Images", columns=4, rows=1)
 
         with gr.Accordion("Advanced options", open=False):
             with gr.Row():