Update app.py

app.py

@@ -6,12 +6,13 @@ from transformers import BertTokenizer, BertForMaskedLM
 
 # 使用私有模型和分詞器
 model_name_or_path = "DeepLearning101/Corrector101zhTW"
+auth_token = os.getenv("HF_HOME") 
 
 # 嘗試加載模型和分詞器
 try:
-    tokenizer = BertTokenizer.from_pretrained(model_name_or_path)
-    model = BertForMaskedLM.from_pretrained(model_name_or_path)
-    model.eval()  # 將模型設置為評估模式
+    tokenizer = BertTokenizer.from_pretrained(model_name_or_path, use_auth_token=auth_token)
+    model = BertForMaskedLM.from_pretrained(model_name_or_path, use_auth_token=auth_token)
+    model.eval()
 except Exception as e:
     print(f"加載模型或分詞器失敗，錯誤信息：{e}")
     exit(1)