app.py

import numpy as np
import gradio as gr
from model import SegmentationTool

seg_tool = SegmentationTool()


def segment(input_img):
    mask_image, transparent_mask_image, image, items, room = seg_tool.get_mask(image=input_img)
    return mask_image


demo = gr.Interface(fn=segment,
                    inputs=gr.Image(type='pil'),
                    outputs=['image'],
                    allow_flagging='never')
if __name__ == "__main__":
    demo.launch()