Update app.py

app.py

@@ -21,8 +21,8 @@ def get_prompt(prompt):
 
 def generate_images(prompt):
     print('calling diffusion lens')
-    image = get_images(prompt, skip_layers=1)
-    return image
+    images = get_images(prompt, skip_layers=1)
+    return images[0]