from transformers import pipeline
import sys
from PIL import Image

def remove_background(image_path, save_path):
    pipe = pipeline("image-segmentation", model="briaai/RMBG-1.4", trust_remote_code=True)
    pillow_image = pipe(image_path) 
    pillow_image.save(save_path)

if __name__ == "__main__":
    image_path = sys.argv[1]
    save_path = sys.argv[2]
    remove_background(image_path, save_path)
    print(f"Image saved to {save_path}")