add local_files_only

model_utils.py

@@ -81,14 +81,14 @@ def load_models():
     return bert_model, bilstm_model
 
 def load_custom_model(model_dir, tokenizer_dir, id2label):
-    config = AutoConfig.from_pretrained(model_dir)
+    config = AutoConfig.from_pretrained(model_dir, local_files_only=True)
     config.id2label = id2label
     config.num_labels = len(id2label)
 
     model = BiLSTMForTokenClassification(model_name=config._name_or_path, num_labels=config.num_labels)
     model.config.id2label = id2label
     model.load_state_dict(torch.load(os.path.join(model_dir, 'pytorch_model.bin'), map_location=torch.device('cpu')))
-    tokenizer = AutoTokenizer.from_pretrained(tokenizer_dir)
+    tokenizer = AutoTokenizer.from_pretrained(tokenizer_dir, local_files_only = True)
 
     return model, tokenizer