Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import torch
+import spaces
 import gradio as gr
 from diffusers import DiffusionPipeline, EulerDiscreteScheduler
 
@@ -23,12 +24,12 @@ pipe = DiffusionPipeline.from_pretrained(
 # KSampler → choose a scheduler (Euler is close to your Comfy euler/simple)
 pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
 pipe.to(device)
-
+spaces.aoti_blocks_load(pipe.transformer.layers, "zerogpu-aoti/Z-Image", variant="fa3")
 
 # ---------------------------------------------------------------------
 # Inference function (maps to CLIPTextEncode + EmptySD3LatentImage + KSampler + VAEDecode)
 # ---------------------------------------------------------------------
-@spaces.GPU(duration=120)
+@spaces.GPU
 def generate_images(
     positive: str,
     negative: str,