app.py

import gradio as gr


def ui_interface(input_image):
    segmented = watershed_segmentation(input_image)
    return segmented

iface = gr.Interface(
    fn=ui_interface,
    inputs=gr.inputs.Image(),
    outputs=gr.outputs.Image()
)

iface.launch()