import argparse from traiNNer.utils.download_util import download_file_from_google_drive if __name__ == "__main__": parser = argparse.ArgumentParser() parser.add_argument("--id", type=str, help="File id") parser.add_argument("--output", type=str, help="Save path") args = parser.parse_args() download_file_from_google_drive(args.id, args.save_path)