Update app.py

app.py

@@ -21,7 +21,8 @@ pipe = DiffusionPipeline.from_pretrained(
     attn_implementation="kernels-community/vllm-flash-attn3",
 )
 
-pipe.transformer.layers._repeated_blocks = ["ZImageTransformerBlock"]
+#pipe.transformer.layers._repeated_blocks = ["ZImageTransformerBlock"]
+#spaces.aoti_blocks_load(pipe.transformer.layers, "zerogpu-aoti/Z-Image", variant="fa3")
 
 @spaces.GPU
 def generate_image(prompt, height, width, num_inference_steps, seed, randomize_seed):
@@ -38,6 +39,8 @@ def generate_image(prompt, height, width, num_inference_steps, seed, randomize_s
         num_inference_steps=int(num_inference_steps),
         guidance_scale=0.0,  # Guidance should be 0 for Turbo models
         generator=generator,
+        num_images_per_prompt=2
+        
     ).images[0]
     
     return image, seed