Update app.py

app.py

@@ -2,13 +2,18 @@ import gradio as gr
 import spaces
 import torch
 from diffusers import AuraFlowPipeline
+import random
 
 pipe = AuraFlowPipeline.from_pretrained("purplesmartai/pony-v7-base", torch_dtype=torch.float16)
 pipe = pipe.to("cuda")
 
 @spaces.GPU()
 def generate_image(prompt, negative_prompt, height, width, num_inference_steps, guidance_scale, seed):
-    generator = torch.Generator("cuda").manual_seed(seed)
+    if seed < 0:
+        seed = random.randint(0, 2**32 - 1)
+
+    generator = torch.Generator("cuda").manual_seed(int(seed))
+
     image = pipe(
         prompt=prompt,
         negative_prompt=negative_prompt,
@@ -16,22 +21,25 @@ def generate_image(prompt, negative_prompt, height, width, num_inference_steps,
         width=int(width),
         num_inference_steps=int(num_inference_steps),
         guidance_scale=guidance_scale,
-        generator=generator
+        generator=generator,
     ).images[0]
-    return image
+    return image, seed
 
 iface = gr.Interface(
     fn=generate_image,
     inputs=[
-        gr.Textbox(label="Prompt", value="A cat holding a sign that says hello world"),
-        gr.Textbox(label="Negative Prompt", placeholder="Enter prompts to exclude"),
+        gr.Textbox(label="Prompt", value="Score_9, "),
+        gr.Textbox(label="Negative Prompt", value="score_6, score_5, score_4, worst quality, low quality, text, deformed, bad hand, blurry, (watermark), extra hands, long ears, ugly, deformed joints, deformed hands, empty background, big ears, narrow face, glowing eyes,", placeholder="Enter prompts to exclude"),
         gr.Slider(label="Height", minimum=256, maximum=2048, step=64, value=1024),
         gr.Slider(label="Width", minimum=256, maximum=2048, step=64, value=1024),
-        gr.Slider(label="Number of Inference Steps", minimum=1, maximum=100, step=1, value=50),
-        gr.Slider(label="Guidance Scale", minimum=1.0, maximum=20.0, step=0.1, value=5.0),
-        gr.Number(label="Seed", value=42)
+        gr.Slider(label="Number of Inference Steps", minimum=1, maximum=100, step=1, value=30),
+        gr.Slider(label="Guidance Scale", minimum=1.0, maximum=20.0, step=0.1, value=3.5),
+        gr.Number(label="Seed (set to -1 for random)", value=-1, minimum=-1)
+    ],
+    outputs=[
+        gr.Image(label="Generated Image", format="png"),
+        gr.Number(label="Used Seed")
     ],
-    outputs=gr.Image(label="Generated Image", format="png"),
     title="purplesmartai/pony-v7-base",
     description="Generate images from text prompts using the AuraFlow model."
 )