Update model_utils.py

model_utils.py

@@ -63,23 +63,6 @@ class BiLSTMForTokenClassification(nn.Module):
 
         return {'loss': loss, 'logits': logits}
 
-# Load custom BiLSTM and pre-trained BERT
-def load_models():
-    """
-    Loads the pre-trained BERT model from Hugging Face Hub.
-
-    Returns:
-        bert_model: The loaded BERT model.
-    """
-    bert_model = AutoModelForTokenClassification.from_pretrained(".models/bert-model")
-    bert_model.eval()
-
-    with open('models/bilstm-model.pkl', 'rb') as f:
-        bilstm_model = pickle.load(f)
-    bilstm_model.eval()
-
-    return bert_model, bilstm_model
-
 def load_custom_model(model_dir, tokenizer_dir, id2label):
     config = AutoConfig.from_pretrained(model_dir, local_files_only=True)
     config.id2label = id2label