import sys
from rembg import remove
from PIL import Image
import io

def main():
    # Read input image bytes from stdin
    input_bytes = sys.stdin.buffer.read()

    # Remove background
    output_bytes = remove(input_bytes)

    # Write output image bytes to stdout
    sys.stdout.buffer.write(output_bytes)

if __name__ == "__main__":
    main()
