DiffTumor / uploading_files_to_hugging_face.py
MrGiovanni's picture
2c4f1a0b6db80fb74def83afdbae8fd1ab70e2d74a2dfbe792fd7b1e0a21e42c
d76218c verified
raw
history blame
1.39 kB
'''
python uploading_files_to_hugging_face.py --folder_directory /path/to/your/folder --repository MrGiovanni/AbdomenAtlasInternal --repository_type dataset
'''
import huggingface_hub
import argparse
def main(args):
huggingface_hub.upload_folder(
folder_path=args.folder_directory,
repo_id=args.repository,
repo_type=args.repository_type,
multi_commits=True,
multi_commits_verbose=True
)
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument('--folder_directory',
dest='folder_directory',
type=str,
default='/Users/zongwei.zhou/Desktop/AbdomenAtlasInternal',
help='the directory of the AbdomenAtlas dataset',
)
parser.add_argument('--repository',
dest='repository',
type=str,
default='MrGiovanni/AbdomenAtlasInternal',
help='repository directory',
)
parser.add_argument('--repository_type',
dest='repository_type',
type=str,
default='dataset',
help='the type of repository',
)
args = parser.parse_args()
main(args)