Update app.py

app.py

@@ -310,7 +310,7 @@ with gr.Blocks(css=css) as demo:
                     preview_image = gr.Image(label="Mask preview")
 
     def use_output_as_input(output_image):
-        return gr.update(value=output_image[1])
+        return output_image
 
     use_as_input_button.click(
         fn=use_output_as_input,