Update app.py

app.py

@@ -21,15 +21,11 @@ def get_prompt(prompt):
 
 def generate_images(prompt):
     print('calling diffusion lens')
-    all_images = []
-    for skip_layers in range(2):
+    yield 'Generating image...'  # Inform the user that the image is being generated
+    for skip_layers in range(23, 0, -1):
         images = get_images(prompt, skip_layers=skip_layers)
-        all_images.append(images[0])
-    print('after call images')
-    return all_images
-
-
+        yield gr.Image(images[0])  # Yield each image as soon as it's ready
 
 with gr.Blocks() as demo:
-    text_input = gr.Interface(fn=generate_images, inputs="text", outputs=["image"]*2)
+    text_input = gr.Interface(fn=generate_images, inputs="text", outputs=["image"] * 23)
 demo.launch()