Update app.py

app.py

@@ -11,13 +11,13 @@ pipe_text2img = LDMZhTextToImagePipeline.from_pretrained(model_id, use_auth_toke
 pipe_text2img = pipe_text2img.to(device)
 
 def infer_text2img(prompt, guide, steps):
-    output = pipe_text2img([prompt]*9, guidance_scale=guide, num_inference_steps=steps)
+    output = pipe_text2img([prompt]*4, guidance_scale=guide, num_inference_steps=steps)
     images = output.images
-    images = [np.array(images[i]) for i in range(9)]
+    images = [np.array(images[i]) for i in range(4)]
     images = np.concatenate([
-        np.concatenate(images[0:3], axis=0),
-        np.concatenate(images[3:6], axis=0),
-        np.concatenate(images[6:9], axis=0),
+        np.concatenate(images[0:2], axis=0),
+        np.concatenate(images[2:4], axis=0),
+    #   np.concatenate(images[6:9], axis=0),
     ], axis=1)
     images = Image.fromarray(images)
     return images