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()