Update model_utils.py

Add id2label as a return value in load_custom_model

model_utils.py

@@ -90,7 +90,7 @@ def load_custom_model(model_dir, tokenizer_dir, 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, local_files_only = True)
 
-    return model, tokenizer
+    return model, tokenizer, id2label
 
 ner_model_dir = "./models/bilstm_ner"
 tokenizer_dir = "./models/tokenizer"