import gradio as gr

def remove_background(image):
    # 배경 제거 로직 구현
    return image

with gr.Blocks() as demo:
    with gr.Row():
        image_input = gr.Image()  # 'tool' 인자 제거
        image_output = gr.Image()
    
    image_input.change(fn=remove_background, inputs=image_input, outputs=image_output)

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