Nechba commited on
Commit
79c2b03
Β·
verified Β·
1 Parent(s): 9ecc265

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -6,7 +6,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStream
6
  import os
7
  from threading import Thread
8
 
9
-
10
  MODEL_ID = "THUDM/glm-4v-9b"
11
 
12
  TITLE = f'<br><center>πŸš€ Coin Generative Recognition</a></center>'
@@ -48,10 +48,11 @@ model = AutoModelForCausalLM.from_pretrained(
48
  low_cpu_mem_usage=True,
49
  trust_remote_code=True
50
  ).to(0)
51
- model.push_to_hub("Nechba/Coin-Generative-Recognition")
 
52
 
53
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, trust_remote_code=True)
54
- tokenizer.push_to_hub("Nechba/Coin-Generative-Recognition")
55
  model.eval()
56
 
57
 
 
6
  import os
7
  from threading import Thread
8
 
9
+ HF_TOKEN = os.environ.get("HF_TOKEN", None)
10
  MODEL_ID = "THUDM/glm-4v-9b"
11
 
12
  TITLE = f'<br><center>πŸš€ Coin Generative Recognition</a></center>'
 
48
  low_cpu_mem_usage=True,
49
  trust_remote_code=True
50
  ).to(0)
51
+ HF_TOKEN = os.environ.get("HF_API", None)
52
+ model.push_to_hub("Nechba/Coin-Generative-Recognition", use_auth_token=HF_TOKEN)
53
 
54
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, trust_remote_code=True)
55
+ tokenizer.push_to_hub("Nechba/Coin-Generative-Recognition",use_auth_token=HF_TOKEN)
56
  model.eval()
57
 
58