| import os | |
| import huggingface_hub | |
| from huggingface_hub import create_branch | |
| #get user variables | |
| repo = input("Repository name: ") | |
| branch = input("New branch name: ") | |
| #login | |
| huggingface_hub.login(os.environ['HF_TOKEN']) | |
| #create the branch | |
| create_branch(f"{repo}", repo_type="model", branch=f"{branch}") | |