import gradio as gr
import imageio

def dummy(img):
  imageio.imwrite("output_image.png", img["mask"])
  return img["image"], img["mask"]

with gr.Blocks() as demo:
  with gr.Row():
    img = gr.Image(tool="sketch", label="base image", show_label=True)
    img1 = gr.Image()
    img2 = gr.Image(label="mask image", show_label=True)
  btn = gr.Button()
  btn.click(dummy, img, [img1, img2])

demo.launch(debug=True)