| """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() | |