Update app.py

app.py

@@ -4,11 +4,13 @@ import torch
 import os
 from transformers import BertTokenizer, BertForMaskedLM
 
+model_name_or_path = "DeepLearning101/Corrector101zhTW"
+auth_token = os.getenv("Corrector") 
 
 # 嘗試加載模型和分詞器
 try:
-    tokenizer = BertTokenizer.from_pretrained("DeepLearning101/Corrector101zhTW", "Corrector")
-    model = BertForMaskedLM.from_pretrained("DeepLearning101/Corrector101zhTW", "Corrector")
+    tokenizer = BertTokenizer.from_pretrained(model_name_or_path, token=auth_token)
+    model = BertForMaskedLM.from_pretrained(model_name_or_path, token=auth_token)
     model.eval()
 except Exception as e:
     print(f"加載模型或分詞器失敗，錯誤信息：{e}")