Update app.py

app.py

@@ -12,6 +12,7 @@ pipe = StableDiffusionPipeline.from_pretrained(model_path, use_safetensors=True,
 pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
 
 g_cuda = torch.Generator()
+g_cuda.manual_seed(g_seed)
 
 def inference(prompt, negative_prompt, num_samples, height=512, width=512, num_inference_steps=50, guidance_scale=7.5, g_seed=52362):
     with torch.inference_mode():
@@ -20,7 +21,7 @@ def inference(prompt, negative_prompt, num_samples, height=512, width=512, num_i
                 negative_prompt=negative_prompt,
                 num_images_per_prompt=int(num_samples),
                 num_inference_steps=int(num_inference_steps), guidance_scale=guidance_scale,
-                generator=g_cuda.manual_seed(g_seed)
+                generator=g_cuda
             ).images
 
 with gr.Blocks() as demo:
@@ -35,7 +36,7 @@ with gr.Blocks() as demo:
             with gr.Row():
                 height = gr.Number(label="Height", value=512)
                 width = gr.Number(label="Width", value=512)
-            num_inference_steps = gr.Slider(label="Steps", value=50)
+            num_inference_steps = gr.Slider(label="Steps", value=25)
         with gr.Column():
             gallery = gr.Gallery()