Updated app with random seed and 25 fixed steps

app.py

@@ -3,8 +3,9 @@ import torch
 from diffusers import FluxPipeline
 from huggingface_hub import HfApi
 import spaces
+import random
 
-@spaces.GPU(duration=70)  # Allocate GPU for 70 seconds
+@spaces.GPU(duration=70)
 def initialize_model():
     model_id = "Freepik/flux.1-lite-8B-alpha"
     pipe = FluxPipeline.from_pretrained(
@@ -17,14 +18,16 @@ def initialize_model():
 def generate_image(
     prompt,
     guidance_scale=3.5,
-    num_steps=28,
-    seed=11,
+    num_steps=25,  # Changed default to 25
     width=1024,
     height=1024
 ):
     # Initialize model within the GPU context
     pipe = initialize_model()
     
+    # Generate random seed
+    seed = random.randint(1, 1000000)
+    
     with torch.inference_mode():
         image = pipe(
             prompt=prompt,
@@ -43,17 +46,15 @@ demo = gr.Interface(
     inputs=[
         gr.Textbox(label="Prompt", placeholder="Enter your image description here..."),
         gr.Slider(minimum=1, maximum=20, value=3.5, label="Guidance Scale", step=0.5),
-        gr.Slider(minimum=1, maximum=50, value=28, label="Number of Steps", step=1),
-        gr.Slider(minimum=1, maximum=1000000, value=11, label="Seed", step=1),
         gr.Slider(minimum=128, maximum=1024, value=1024, label="Width", step=64),
         gr.Slider(minimum=128, maximum=1024, value=1024, label="Height", step=64)
     ],
     outputs=gr.Image(type="pil", label="Generated Image"),
     title="Flux Image Generator (Zero-GPU)",
-    description="Generate images using Freepik's Flux model with Zero-GPU allocation",
+    description="Generate images using Freepik's Flux model with Zero-GPU allocation. Using 25 steps and random seed for each generation.",
     examples=[
-        ["A close-up image of a green alien with fluorescent skin in the middle of a dark purple forest", 3.5, 28, 11, 1024, 1024],
-        ["A serene landscape with mountains at sunset", 3.5, 28, 42, 1024, 1024],
+        ["A close-up image of a green alien with fluorescent skin in the middle of a dark purple forest", 3.5, 1024, 1024],
+        ["A serene landscape with mountains at sunset", 3.5, 1024, 1024],
     ]
 )