Update app.py

app.py

@@ -3,16 +3,17 @@
 import gradio as gr
 import operator
 import torch
+import os
 from transformers import BertTokenizer, BertForMaskedLM
 
 # 使用私有模型和分詞器
 model_name_or_path = "DeepLearning101/Corrector101zhTW"
 auth_token = os.getenv("Corrector101zhTW")  # 從環境變量中獲取 token
 
-# 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)
-tokenizer = BertTokenizer.from_pretrained(model_name_or_path)
-model = BertForMaskedLM.from_pretrained(model_name_or_path)
+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)
+# tokenizer = BertTokenizer.from_pretrained(model_name_or_path)
+# model = BertForMaskedLM.from_pretrained(model_name_or_path)
 
 def ai_text(text):
     with torch.no_grad():