Update app.py

app.py

@@ -17,11 +17,11 @@ sd_dreambooth_model._diffusion_model = db_diffusion_model
 # generate images
 def infer(prompt):
     generated_images = sd_dreambooth_model.text_to_image(
-        prompt, batch_size=9
+        prompt, batch_size=4
     )
     return generated_images 
     
-output = gr.Gallery(label="Outputs").style(grid=(3,3))
+output = gr.Gallery(label="Outputs").style(grid=(2,2))
 
 # customize interface
 title = "Dreambooth Demo on Dog Images"