Update app.py

app.py

@@ -23,7 +23,8 @@ def generate_images(prompt):
     print('calling diffusion lens')
     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
+        yield images[0]  # Yield each image as soon as it's ready
+        # 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")