Update app.py

app.py

@@ -6,12 +6,12 @@ from transformers import BertTokenizer, BertForMaskedLM
 
 # 使用私有模型和分詞器
 model_name_or_path = "DeepLearning101/Corrector101zhTW"
-auth_token = os.getenv("HF_HOME") 
+auth_token = os.getenv("Corrector") 
 
 # 嘗試加載模型和分詞器
 try:
-    tokenizer = BertTokenizer.from_pretrained(model_name_or_path, use_auth_token=True, token=auth_token)
-    model = BertForMaskedLM.from_pretrained(model_name_or_path, use_auth_token=True, token=auth_token)
+    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}")