Update app.py

Set back before AIoT

app.py

@@ -175,7 +175,7 @@ def generate_image(
     pipe,
     prompt,
     resolution="1024x1024",
-    seed=42,
+    seed=-1,
     guidance_scale=5.0,
     num_inference_steps=50,
     shift=3.0,
@@ -184,6 +184,10 @@ def generate_image(
 ):
     width, height = get_resolution(resolution)
 
+    if seed == -1:
+        seed = torch.randint(0, 1000000, (1,)).item()
+    print(f"Using seed: {seed}")
+
     generator = torch.Generator("cuda").manual_seed(seed)
 
     scheduler = FlowMatchEulerDiscreteScheduler(num_train_timesteps=1000, shift=shift)
@@ -400,16 +404,11 @@ def generate(
         gallery_images = []
     gallery_images.append(image)
 
-    return gallery_images, str(new_seed), str(new_seed)
+    return gallery_images, str(new_seed)
 
 
 init_app()
 
-# ==================== AoTI (Ahead of Time Inductor compilation) ====================
-
-pipe.transformer.layers._repeated_blocks = ["ZImageTransformerBlock"]
-spaces.aoti_blocks_load(pipe.transformer.layers, "zerogpu-aoti/Z-Image", variant="fa3")
-
 with gr.Blocks(title="Z-Image Demo") as demo:
     gr.Markdown(
         """<div align="center">
@@ -439,7 +438,7 @@ with gr.Blocks(title="Z-Image Demo") as demo:
                 resolution = gr.Dropdown(value=initial_res_choices[0], choices=initial_res_choices, label="Width x Height (Ratio)")
 
             with gr.Row():
-                seed = gr.Number(label="Seed", value=42, precision=0)
+                seed = gr.Number(label="Seed", value=-1, precision=0)
                 random_seed = gr.Checkbox(label="Random Seed", value=True)
 
             with gr.Row():
@@ -477,14 +476,23 @@ with gr.Blocks(title="Z-Image Demo") as demo:
     # Dummy enable_enhance variable set to False
     enable_enhance = gr.State(value=False)
 
+    def update_seed(current_seed, random_seed_enabled):
+        if random_seed_enabled:
+            new_seed = random.randint(1, 1000000)
+        else:
+            new_seed = current_seed if current_seed != -1 else random.randint(1, 1000000)
+        return gr.update(value=new_seed)
+
+    generate_btn.click(update_seed, inputs=[seed, random_seed], outputs=[seed])
+
     generate_btn.click(
         generate,
         inputs=[prompt_input, resolution, seed, steps, shift, enable_enhance, random_seed, output_gallery],
-        outputs=[output_gallery, used_seed, seed],
+        outputs=[output_gallery, used_seed],
     )
 
 css='''
 .fillable{max-width: 1230px !important}
 '''
 if __name__ == "__main__":
-    demo.launch(css=css)
+    demo.launch(css=css)