i

main.py

@@ -23,6 +23,10 @@ def example(_image):
     pass
 
 
+def send_to_input(output):
+    return output
+
+
 def ui():
     with gr.Blocks() as app:
         with gr.Row():
@@ -44,12 +48,16 @@ def ui():
 
             with gr.Column():
                 output_image = gr.Image(type="pil", label="Output Image")
+                send_to_input_btn = gr.Button(value="Use as input", variant="secondary")
 
         gr.Markdown(
             "The lllyasviel's original repo is [here](https://github.com/lllyasviel/AdverseCleaner/tree/main)."
         )
 
         start_btn.click(fn=clean_image, inputs=[input_image], outputs=[output_image])
+        send_to_input_btn.click(
+            fn=send_to_input, inputs=[output_image], outputs=[input_image]
+        )
 
     app.launch()