change 10 images to 4 images

app.py

@@ -17,7 +17,7 @@ pipe = pipe.to(device)
 model, preprocess = clip.load("ViT-B/32", device=device)
 
 def generate_image(text_prompt):
-    images = pipe(text_prompt, num_images_per_prompt=10).images
+    images = pipe(text_prompt, num_images_per_prompt=4).images
     return images
 
 def build_generation_block(prompt):
@@ -34,7 +34,7 @@ def build_generation_block(prompt):
 
     gallery = gr.Gallery(
         label="Generated images", show_label=False, elem_id="gallery"
-    ).style(columns=[5], rows=[2], object_fit="contain", height="auto")
+    ).style(columns=[2], rows=[2], object_fit="contain", height="auto")
     
     btn.click(generate_image, text, gallery)