Update app.py

app.py

@@ -21,20 +21,11 @@ def get_prompt(prompt):
 
 def generate_images(prompt):
     print('calling diffusion lens')
-    images = []
-    for skip_layers in range(1):  # loop from 0 to 23
-        image = get_images(prompt, skip_layers=skip_layers)
-        images.append(image)
-
-    return images
+    image = get_images(prompt, skip_layers=skip_layers)
+    return image
 
 
 
 with gr.Blocks() as demo:
-    # gallery = gr.Gallery(
-    #     label="Generated images", show_label=False, elem_id="gallery",
-    #     columns=[6], rows=[4], object_fit="contain", height="auto")  # set rows to 24 to accommodate all images
-    # btn = gr.Button("Generate images", scale=0)
-    text_input = gr.Interface(fn=generate_images, inputs="text", outputs=["image"] * 1)
-    # btn.click(generate_images, text_input, gallery)  # pass the text input interface to btn.click()
+    text_input = gr.Interface(fn=generate_images, inputs="text", outputs="image")
 demo.launch()