Update app.py

app.py

@@ -21,12 +21,15 @@ def get_prompt(prompt):
 
 def generate_images(prompt):
     print('calling diffusion lens')
-    images = get_images(prompt, skip_layers=1)
+    all_images = []
+    for skip_layers in range(2):
+        images = get_images(prompt, skip_layers=skip_layers)
+        all_images.append(images[0])
     print('after call images')
-    return images[0]
+    return all_images
 
 
 
 with gr.Blocks() as demo:
-    text_input = gr.Interface(fn=generate_images, inputs="text", outputs="image")
+    text_input = gr.Interface(fn=generate_images, inputs="text", outputs=["image"]*2)
 demo.launch()