Update app.py

app.py

@@ -12,8 +12,16 @@ model_name_or_path = "DeepLearning101/Corrector101zhTW"
 
 # 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)
+# model = BertForMaskedLM.from_pretrained(model_name_or_path)
+
+model_name_or_path = "DeepLearning101/Corrector101zhTW"
+
+try:
+    tokenizer = BertTokenizer.from_pretrained(model_name_or_path)
+    model = BertForMaskedLM.from_pretrained(model_name_or_path)
+except Exception as e:
+    
 
 def ai_text(text):
     with torch.no_grad():