no need label

app.py

@@ -32,8 +32,12 @@ def display_images(images):
 
 def get_prompt(prompt):
     print('prompt:', prompt)
-    print('calling diffusion lens')
+    return prompt
+
 
+
+def generate_images(prompt):
+    print('calling diffusion lens')
     for skip_layers in range(24):  # loop from 0 to 23
         image = get_images(prompt, skip_layers=skip_layers)
         gallery.update(image)  # update the gallery with the new image
@@ -47,6 +51,6 @@ with gr.Blocks() as demo:
         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=get_prompt, inputs="text", outputs="text")
-    btn.click(get_prompt, "text", gallery)  # pass the text input interface to btn.click()
+    text_input = gr.Interface(fn=get_prompt, inputs="text", outputs="text")
+    btn.click(generate_images, text_input, gallery)  # pass the text input interface to btn.click()
 demo.launch()