Add seed support

app.py

@@ -2,10 +2,14 @@ import gradio as gr
 import torch
 from diffusers import DiffusionPipeline
 
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
 def generate(
-    prompt, negative_prompt, num_inference_steps, width, height, guidance_scale
+    prompt, negative_prompt, num_inference_steps, width, height, guidance_scale, seed
 ):
-    pipeline = DiffusionPipeline.from_pretrained("Lykon/DreamShaper")
+    pipeline = DiffusionPipeline.from_pretrained("Lykon/DreamShaper", safety_checker=None)
+    pipeline = pipeline.to(device)
+    generator = torch.Generator(device=device).manual_seed(seed)
 
     return pipeline(
         prompt=prompt,
@@ -14,6 +18,7 @@ def generate(
         width=width,
         height=height,
         guidance_scale=guidance_scale,
+        generator=generator,
     ).images[0]
 
 iface = gr.Interface(
@@ -25,6 +30,13 @@ iface = gr.Interface(
         gr.Slider(label="Width", minimum=64, maximum=2048, value=512, step=1),
         gr.Slider(label="Height", minimum=64, maximum=2048, value=512, step=1),
         gr.Slider(label="CFG Scale", minimum=1, maximum=30, value=9, step=0.5),
+        gr.Slider(
+            label="Seed",
+            minimum=0,
+            maximum=4294967294,
+            step=1,
+            randomize=True,
+        ),
     ],
     outputs="image",
 )