File size: 898 Bytes
4f03424 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | """Run interactively in your own terminal; never pass tokens on the command line."""
import argparse
from getpass import getpass
def main():
ap=argparse.ArgumentParser()
ap.add_argument('platform',choices=('huggingface','modelscope'))
args=ap.parse_args()
if args.platform=='huggingface':
from huggingface_hub import login,HfApi
login(token=getpass('Hugging Face write token (hidden): '),add_to_git_credential=False)
owner=HfApi().whoami()['name']
expected='ixim'
else:
from modelscope_hub import HubApi
api=HubApi()
user=api.login(getpass('ModelScope SDK write token (hidden): '))
owner=user.username
expected='iximbox'
if owner!=expected: raise ValueError(f'Expected {expected}; logged in as {owner}')
print(f'Authenticated {owner}. No token has been printed.')
if __name__=='__main__': main()
|