| import gradio as gr | |
| def draw_mask(img, h, w): | |
| return [ | |
| img['image'].convert("RGB").resize((h, w)), | |
| img['mask'].convert("RGB").resize((h, w)) | |
| ] | |
| iface = gr.Interface( | |
| fn=draw_mask, | |
| inputs=[ | |
| gr.Image(source="upload", tool='sketch', type='pil'), | |
| gr.Slider(label="height", minimum=256, maximum=1024, value=512), | |
| gr.Slider(label="width", minimum=256, maximum=1024, value=512), | |
| ], outputs=[ | |
| gr.Image(), | |
| gr.Image(), | |
| ]) | |
| iface.launch() |