Update app_merged.py

app_merged.py

@@ -1511,6 +1511,14 @@ with gr.Blocks() as app:
             
             with gr.Column():
                 output_image.render()
+                transfer_btn = gr.Button("Send to relight")
+
+        def send_img(img_result):
+                return img_result
+        
+        transfer_btn.click(send_img, [output_image], [input_fg])
+
+                
         generate_btn.click(
             fn=generate_image,
             inputs=[prompt_input, structure_image, style_image, depth_strength, style_strength],