DeepLearning101 commited on
Commit
0ab3e57
·
1 Parent(s): 5acee80

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -8,12 +8,12 @@ from transformers import BertTokenizer, BertForMaskedLM
8
 
9
  # 使用私有模型和分詞器
10
  model_name_or_path = "DeepLearning101/Corrector101zhTW"
11
- auth_token = os.getenv("Corrector101zhTW") # 從環境變量中獲取 token
12
 
13
- tokenizer = BertTokenizer.from_pretrained(model_name_or_path, use_auth_token=auth_token)
14
- model = BertForMaskedLM.from_pretrained(model_name_or_path, use_auth_token=auth_token)
15
- # tokenizer = BertTokenizer.from_pretrained(model_name_or_path)
16
- # model = BertForMaskedLM.from_pretrained(model_name_or_path)
17
 
18
  def ai_text(text):
19
  with torch.no_grad():
 
8
 
9
  # 使用私有模型和分詞器
10
  model_name_or_path = "DeepLearning101/Corrector101zhTW"
11
+ # auth_token = os.getenv("Corrector101zhTW") # 從環境變量中獲取 token
12
 
13
+ # tokenizer = BertTokenizer.from_pretrained(model_name_or_path, use_auth_token=auth_token)
14
+ # model = BertForMaskedLM.from_pretrained(model_name_or_path, use_auth_token=auth_token)
15
+ tokenizer = BertTokenizer.from_pretrained(model_name_or_path)
16
+ model = BertForMaskedLM.from_pretrained(model_name_or_path)
17
 
18
  def ai_text(text):
19
  with torch.no_grad():