Update app.py

app.py

@@ -4,14 +4,11 @@ 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(model_name_or_path, token=auth_token)
-    model = BertForMaskedLM.from_pretrained(model_name_or_path, token=auth_token)
+    tokenizer = BertTokenizer.from_pretrained("DeepLearning101/Corrector101zhTW", "Corrector")
+    model = BertForMaskedLM.from_pretrained("DeepLearning101/Corrector101zhTW", "Corrector")
     model.eval()
 except Exception as e:
     print(f"加載模型或分詞器失敗，錯誤信息：{e}")