Update app.py

app.py

@@ -1,14 +1,20 @@
 import torch
 import spaces
 import gradio as gr
-from diffusers import DiffusionPipeline
+from diffusers import ZImagePipeline, ZImageTransformer2DModel
 
 # Load the pipeline once at startup
 print("Loading Z-Image-Turbo pipeline...")
-pipe = DiffusionPipeline.from_pretrained(
+transformer = ZImageTransformer2DModel.from_pretrained(
+    "linoyts/beyond-reality-z-image-diffusers",
+    torch_dtype=torch.bfloat16,
+)
+
+# Load pipeline with custom transformer
+pipe = ZImagePipeline.from_pretrained(
     "Tongyi-MAI/Z-Image-Turbo",
+    transformer=transformer,
     torch_dtype=torch.bfloat16,
-    low_cpu_mem_usage=False,
 )
 pipe.to("cuda")
 
@@ -65,8 +71,8 @@ with gr.Blocks(fill_height=True) as demo:
     # Header
     gr.Markdown(
         """
-        # 🎨 Z-Image-Turbo
-        **Ultra-fast AI image generation** • Generate stunning images in just 8 steps
+        # Z-Image-Turbo BEYOND REALITY
+        **Ultra-fast AI image generation** with [Nurburgring/BEYOND_REALITY_Z_IMAGE](https://huggingface.co/Nurburgring/BEYOND_REALITY_Z_IMAGE) - Z Image Turbo fine-tuned for enhanced texture fidelity, analog photography alignment & improved balance of realism & aesthetics.
         """,
         elem_classes="header-text"
     )
@@ -104,7 +110,7 @@ with gr.Blocks(fill_height=True) as demo:
                 num_inference_steps = gr.Slider(
                     minimum=1,
                     maximum=20,
-                    value=9,
+                    value=10,
                     step=1,
                     label="Inference Steps",
                     info="9 steps = 8 DiT forwards (recommended)"