Update model.py

model.py

@@ -230,8 +230,10 @@ def predict_location_preds(tokenizer, model, feature_text, pn_history):
     return location_preds, ', '.join(x)
 
 def get_predictions(feature_text, pn_history):
+    feature_text = feature_text.lower().replace("-OR-", ";-").replace("-", " ")
+    pn_history = pn_history.lower()
+    
     location_preds, pred_string = predict_location_preds(tokenizer, model, [feature_text], [pn_history])
-    # print(pred_string)
     return pred_string
 
 tokenizer = AutoTokenizer.from_pretrained(config['tokenizer_path'])