import gradio as gr
from rembg import remove

def remove_background(image):
    output_image = remove(image)
    return output_image

iface = gr.Interface(
    fn=remove_background,
    inputs=gr.Image(type="pil"),
    outputs=gr.Image(type="pil"),
    title="배경 제거 도구",
    description="이 도구를 사용하여 이미지에서 배경을 제거하세요."
)

if __name__ == "__main__":
    iface.launch()