Update app.py

app.py

@@ -2,6 +2,8 @@ import torch
 import spaces
 import gradio as gr
 from diffusers import ZImagePipeline, ZImageTransformer2DModel
+import numpy as np
+import random
 
 # Load the pipeline once at startup
 print("Loading Z-Image-Turbo pipeline...")
@@ -23,12 +25,15 @@ pipe.to("cuda")
 # spaces.aoti_blocks_load(pipe.transformer.layers, "zerogpu-aoti/Z-Image", variant="fa3")
 
 print("Pipeline loaded!")
+MAX_SEED = np.iinfo(np.int32).max
+
+
 
 @spaces.GPU
-def generate_image(prompt, height, width, num_inference_steps, seed, randomize_seed, progress=gr.Progress(track_tqdm=True)):
+def generate_image(prompt, height, width, num_inference_steps=10, seed=42, randomize_seed=True, progress=gr.Progress(track_tqdm=True)):
     """Generate an image from the given prompt."""
     if randomize_seed:
-        seed = torch.randint(0, 2**32 - 1, (1,)).item()
+        seed = random.randint(0, MAX_SEED)
     
     generator = torch.Generator("cuda").manual_seed(int(seed))
     image = pipe(