Update model.py

model.py

@@ -174,9 +174,9 @@ def get_location_predictions(preds, offset_mapping, sequence_ids, test=False):
 
 
 
-def predict_location_preds(tokenizer, model, feature_text, pn_history):
+def predict_location_preds(tokenizer, model, feature_text, pn_history, pn_history_lower):
 
-    test_ds = NBMETestData(feature_text, pn_history, tokenizer)
+    test_ds = NBMETestData(feature_text, pn_history_lower, tokenizer)
     test_dl = torch.utils.data.DataLoader(
         test_ds, 
         batch_size=config['batch_size'], 
@@ -231,9 +231,9 @@ def predict_location_preds(tokenizer, model, feature_text, pn_history):
 
 def get_predictions(feature_text, pn_history):
     feature_text = feature_text.lower().replace("-OR-", ";-").replace("-", " ")
-    pn_history = pn_history.lower()
+    pn_history_lower = pn_history.lower()
     
-    location_preds, pred_string = predict_location_preds(tokenizer, model, [feature_text], [pn_history])
+    location_preds, pred_string = predict_location_preds(tokenizer, model, [feature_text], [pn_history], [pn_history_lower])
     
     if pred_string == "":
         pred_string = 'Feature not present!'