Anthonyg5005 commited on
Commit
a4ebec9
·
verified ·
1 Parent(s): 7bef370

added useful notes

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