"""Upload weights to ModelScope. Usage: MS_TOKEN=xxx python scripts/push_ms.py [namespace/g2pS]"""
import os
import sys

from modelscope.hub.api import HubApi

repo = sys.argv[1] if len(sys.argv) > 1 else 'baicai-1145/g2pS'
api = HubApi()
api.login(os.environ['MS_TOKEN'])
api.create_repo(repo, repo_type='model', exist_ok=True)
# upload weights + meta (skip code dirs)
folders = ['models', 'torch_weights']
files = ['README.md', 'MODEL_CARD.md', 'LICENSE']
for f in files:
    api.upload_file(repo_id=repo, repo_type='model', path_or_fileobj=f, path_in_repo=f)
for d in folders:
    for root, _, fs in os.walk(d):
        for f in fs:
            p = os.path.join(root, f)
            api.upload_file(repo_id=repo, repo_type='model', path_or_fileobj=p, path_in_repo=p)
print('uploaded', repo)