import gradio as gr


def segment_image(input_image):
    output_image = input_image
    return output_image


with gr.Blocks() as demo:

    with gr.Row():
        input_image = gr.Image(label='Input Image')
        output_image = gr.Image(label='Output Image')

    button = gr.Button('Mask Image')
    button.click(segment_image, inputs=[input_image], outputs=output_image)

    gr.Examples(
        examples = [
            ['./imgs/cat.jpg']
        ],
        inputs=[input_image],
        outputs=[output_image],
        fn=segment_image,
        cache_examples=True
    )

demo.launch()