Remove background

remove_bg.py

 import argparse
 from rembg import remove # pip install rembg
 from PIL import Image

 def main():
     parser = argparse.ArgumentParser(description="Remove background from an image using rembg")
     parser.add_argument("input_path", help="Input image file path")
     parser.add_argument("output_path", help="Output image file path")
     args = parser.parse_args()

     input_path = args.input_path
     output_path = args.output_path

     inp = Image.open(input_path)
     output = remove(inp)
     output.save(output_path)
     print(f"Background removed and saved to {output_path}")

 if __name__ == "__main__":
     main()