app.py

import gradio as gr
from PIL import Image, ImageDraw

def generate_masked_image(dict):
    mask = dict["mask"].convert("RGB")
    return mask

iface = gr.Interface(
    fn=generate_masked_image, 
    inputs=[
        gr.inputs.Image(label="Original", type="pil", tool="sketch")
    ], 
    outputs=gr.outputs.Image(label="Mask", type="pil")
)

iface.launch()