Hugging Face's logo

Spaces:
DeepLearning101
/
Corrector101zhTW
Running

App Files Community
DeepLearning101 commited on
Commit
343aee7
·
verified ·
1 Parent(s): 9aa7185

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from transformers import BertTokenizer, BertForMaskedLM
7
 
8
  # 使用私有模型和分詞器
9
  model_name_or_path = "DeepLearning101/Corrector101zhTW"
10
- auth_token = "Corrector101zhTW" # 換成您的 Hugging Face API token
11
 
12
  tokenizer = BertTokenizer.from_pretrained(model_name_or_path, use_auth_token=auth_token)
13
  model = BertForMaskedLM.from_pretrained(model_name_or_path, use_auth_token=auth_token)
 
7
 
8
  # 使用私有模型和分詞器
9
  model_name_or_path = "DeepLearning101/Corrector101zhTW"
10
+ auth_token = os.getenv("Corrector101zhTW") # 從環境變量中獲取 token
11
 
12
  tokenizer = BertTokenizer.from_pretrained(model_name_or_path, use_auth_token=auth_token)
13
  model = BertForMaskedLM.from_pretrained(model_name_or_path, use_auth_token=auth_token)