Anthonyg5005 commited on
Commit
0df50e1
·
verified ·
1 Parent(s): 0808d21

Replaced by Manage Branch.py

Browse files
Files changed (1) hide show
  1. create new branch.py +0 -56
create new branch.py DELETED
@@ -1,56 +0,0 @@
1
- #import required modules
2
- import os
3
- import huggingface_hub
4
- from huggingface_hub import create_branch
5
-
6
- #get user variables
7
- repo = input("Repository name: ")
8
- while True:
9
- r_type = input("Repo type (model) (dataset) (space): ").lower()
10
-
11
- if r_type not in ['model', 'dataset', 'space']:
12
- print("Please choose one of the three options.")
13
- continue
14
-
15
- break
16
- branch = input("New branch name (No spaces): ")
17
-
18
- #get token
19
- if 'HF_TOKEN' in os.environ:
20
- #if the variable is found then write it to hf_token:
21
- hf_token = os.environ['HF_TOKEN']
22
- tfound = "true"
23
- else:
24
- #if the variable is not found then prompt user to provide it:
25
- hf_token = input("HF_TOKEN Variable not detected. Enter your HuggingFace (WRITE) token: ")
26
- tfound = "false"
27
-
28
- #login
29
- huggingface_hub.login(hf_token)
30
-
31
- #create the branch
32
- create_branch(repo, repo_type=r_type, branch=branch)
33
-
34
- #extra information
35
- #won't work if special characters are used
36
- if r_type == 'model':
37
- print(f"Branch created at https://huggingface.co/{repo}/tree/{branch}")
38
- elif r_type == 'dataset':
39
- print(f"Branch created at https://huggingface.co/datasets/{repo}/tree/{branch}")
40
- elif r_type == 'space':
41
- print(f"Branch created at https://huggingface.co/spaces/{repo}/tree/{branch}")
42
- #if token wasn't found then display following text:
43
- if tfound == 'false':
44
- print('''
45
- Set HF_TOKEN to a token with WRITE permissions to skip inputting token on each run.
46
-
47
- On Unix systems, edit the file ~/.bashrc with an editor of your choise.
48
- On a new line add: export HF_TOKEN=Your-HuggingFace-token-here
49
- (Terminal Refresh Required)
50
- To temporarily set a token to the active terminal use 'export HF_TOKEN=Your-HuggingFace-token-here'
51
-
52
- On Windows use 'setx HF_TOKEN Your-HuggingFace-token-here'
53
- (System Restart Required)
54
- To temporarily set a token to the active terminal use 'set HF_TOKEN=Your-HuggingFace-token-here'
55
- ''')
56
- input("Press enter to continue.")