stevhliu's picture
stevhliu HF staff
Create app.py
c10f4a4
raw
history blame
414 Bytes
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)