Update app.py

app.py

@@ -24,6 +24,10 @@ pipe = DiffusionPipeline.from_pretrained(
 #pipe.transformer.layers._repeated_blocks = ["ZImageTransformerBlock"]
 #spaces.aoti_blocks_load(pipe.transformer.layers, "zerogpu-aoti/Z-Image", variant="fa3")
 
+pipe.to("cuda")
+
+
+
 @spaces.GPU
 def generate_image(prompt, height, width, num_inference_steps, seed, randomize_seed):
     """Generate an image from the given prompt."""