DeepLearning101 commited on
Commit
17ecded
·
1 Parent(s): 343aee7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -9,9 +9,10 @@ from transformers import BertTokenizer, BertForMaskedLM
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)
14
-
 
15
 
16
  def ai_text(text):
17
  with torch.no_grad():
 
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)
14
+ tokenizer = BertTokenizer.from_pretrained(model_name_or_path)
15
+ model = BertForMaskedLM.from_pretrained(model_name_or_path)
16
 
17
  def ai_text(text):
18
  with torch.no_grad():