Update app.py

app.py

@@ -21,11 +21,10 @@ def get_prompt(prompt):
 
 def generate_images(prompt):
     print('calling diffusion lens')
-    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)
         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"] * 23)
-demo.launch()
+    text_input = gr.Interface(fn=generate_images, inputs="text", outputs="image")
+demo.launch()