app.py

import gradio as gr
from inference import get_output
from PIL import Image

def predict(image, mask):
    return get_output(image, mask)

iface = gr.Interface(fn=predict, inputs=[gr.Image(type= "numpy"), gr.Image(type= "numpy")], outputs="image")
iface.launch(show_error=True)