Update app.py

app.py

@@ -15,9 +15,11 @@ pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=os.envir
 pipe = pipe.to(device)
 
 def infer(prompt):
+    generator = torch.Generator(device=device)
     with autocast("cuda"):
         images_list = pipe(
-            [prompt])['sample']
+            [prompt],
+        generator=generator)
     return images_list
 
 text = gr.Textbox(