Update app.py

app.py

@@ -10,16 +10,25 @@ pipe = ZImagePipeline.from_pretrained(
     torch_dtype=torch.bfloat16,
     low_cpu_mem_usage=False,
 )
+pipe_no_lora = ZImagePipeline.from_pretrained(
+    "Tongyi-MAI/Z-Image-Turbo",
+    torch_dtype=torch.bfloat16,
+    low_cpu_mem_usage=False,
+)
 pipe.load_lora_weights("bdsqlsz/qinglong_DetailedEyes_Z-Image", weight_name="qinglong_detailedeye_z-imageV2(comfy).safetensors", adapter_name="lora")
-pipe.set_adapters(["lora",], adapter_weights=[1.1])
-pipe.fuse_lora(adapter_names=["lora"], lora_scale=1.)
+pipe.set_adapters(["lora",], adapter_weights=[1.])
+pipe.fuse_lora(adapter_names=["lora"], lora_scale=1.15)
 pipe.unload_lora_weights()
 pipe.to("cuda")
+pipe_no_lora.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_no_lora.transformer.layers._repeated_blocks = ["ZImageTransformerBlock"]
+spaces.aoti_blocks_load(pipe_no_lora.transformer.layers, "zerogpu-aoti/Z-Image", variant="fa3")
+
 print("Pipeline loaded!")
 
 @spaces.GPU
@@ -37,8 +46,17 @@ def generate_image(prompt, height, width, num_inference_steps, seed, randomize_s
         guidance_scale=0.0,  # Guidance should be 0 for Turbo models
         generator=generator,
     ).images[0]
+
+    image_no_lora = pipe_no_lora(
+        prompt=prompt,
+        height=int(height),
+        width=int(width),
+        num_inference_steps=int(num_inference_steps),
+        guidance_scale=0.0,  # Guidance should be 0 for Turbo models
+        generator=generator,
+    ).images[0]
     
-    return image, seed
+    return (image_no_lora,image), seed
 
 
 # Example prompts
@@ -109,7 +127,7 @@ with gr.Blocks(title="Z-Image-Turbo Demo") as demo:
             generate_btn = gr.Button("🚀 Generate", variant="primary", size="lg")
         
         with gr.Column(scale=1):
-            output_image = gr.Image(
+            output_image = gr.ImageSlider(
                 label="Generated Image",
                 type="pil",
             )