Update app.py

app.py

@@ -4,7 +4,9 @@ from PIL import Image
 
 from lambda_diffusers import StableDiffusionImageEmbedPipeline
 
-def ask(input_im, scale, n_samples, steps, seed):
+def ask(input_im, scale, n_samples, steps, seed, images):
+    images = images
+    generator = torch.Generator(device=device).manual_seed(int(seed))
     
     images_list = pipe(
         n_samples*[input_im],
@@ -18,14 +20,15 @@ def ask(input_im, scale, n_samples, steps, seed):
             images.append(safe_image)
         else:
             images.append(image)
+     return images
 
 def main(input_im, scale, n_samples, steps, seed):
     
-    generator = torch.Generator(device=device).manual_seed(int(seed))
+    
 
     images = []
-    ask(input_im, scale, n_samples, steps, seed)
-    ask(input_im, scale, n_samples, steps, seed)
+    images = ask(input_im, scale, n_samples, steps, seed, images)
+    images = ask(input_im, scale, n_samples, steps, seed, images)
     
     return images