Update app.py

app.py

@@ -1,7 +1,7 @@
 import torch
 import spaces
 import gradio as gr
-from diffusers import DiffusionPipeline
+from diffusers import ZImagePipeline
 
 # Load the pipeline once at startup
 print("Loading Z-Image-Turbo pipeline...")
@@ -11,12 +11,12 @@ pipe = DiffusionPipeline.from_pretrained(
     low_cpu_mem_usage=False,
 )
 
-pipe.load_lora_weights("ostris/z_image_turbo_childrens_drawings", weights_name="z_image_turbo_childrens_drawings.safetensors")
+pipe.load_lora_weights("renderartist/Technically-Color-Z-Image-Turbo", weights_name="Technically_Color_Z_Image_Turbo_v1_renderartist_2000.safetensors")
 pipe.to("cuda")
 
 # ======== AoTI compilation + FA3 ========
-pipe.transformer.layers._repeated_blocks = ["ZImageTransformerBlock"]
-spaces.aoti_blocks_load(pipe.transformer.layers, "zerogpu-aoti/Z-Image", variant="fa3")
+# pipe.transformer.layers._repeated_blocks = ["ZImageTransformerBlock"]
+# spaces.aoti_blocks_load(pipe.transformer.layers, "zerogpu-aoti/Z-Image", variant="fa3")
 
 print("Pipeline loaded!")