fix count of images

main.py

@@ -54,9 +54,9 @@ def draw(data: Data):
 		pipe = pipe.to('cuda')
 
 		generator = torch.Generator("cuda").manual_seed(seed)
-		images = pipe(prompt, negative_prompt=n_prompt, guidance_scale=7.5, generator=generator, num_images_per_prompt=4).images
+		images = pipe(prompt, negative_prompt=n_prompt, guidance_scale=7.5, generator=generator, num_images_per_prompt=3).images
 
-		grid = image_grid(images, rows=2, cols=2)
+		grid = image_grid(images, rows=3, cols=1)
 
 		fileName = "sd_" + str(time.time()) + '.png'
 		grid.save("/code/tmpdir/" + fileName)