Update app.py

app.py

@@ -177,16 +177,16 @@ with gr.Blocks(theme=theme, css=css, elem_id="app-container") as app:
                             
             with gr.Row():
                 text_button = gr.Button("Run", variant='primary', elem_id="gen-button")
+            with gr.Row():
+                clr_button =gr.Button("Clear prompt",variant="primary", elem_id="clear_button")
+                clr_button.click(lambda: gr.Textbox(value=""), None, text_prompt)
 
-                
             with gr.Row():
                 image_output = gr.Image(type="pil", label="Image Output", format="png", elem_id="gallery")
                 
             with gr.Row(): 
-                #clear_btn = gr.Button(value="Clear Image",variant="primary", elem_id="clear_button")
-                #clear_btn.click(clear, inputs=[ ], outputs=[image_output])
-                clr_button =gr.Button("Clear",variant="primary", elem_id="clear_button")
-                clr_button.click([lambda: gr.Textbox(value=""), None, text_prompt], [lambda: gr.Image(value=""), None, image_output])
+                clear_btn = gr.Button(value="Clear Image",variant="primary", elem_id="clear_button")
+                clear_btn.click(clear, inputs=[ ], outputs=[image_output])