Update predictor.py

predictor.py

@@ -193,8 +193,15 @@ def predict_srl_allennlp_like_spacy(
 
     return words, results
 
+def normalize_whitespace(s: str) -> str:
+    if s is None:
+        return ""
+    # strip leading/trailing spaces (incl. non-breaking etc.)
+    s = s.replace("\u00A0", " ").replace("\u2009", " ").strip()
+    return s
 
 def main_predictor(model_path, bert_name, sentence, spacy_model="en_core_web_md"):
+    sentence = normalize_whitespace(sentence)
     device = "cuda" if torch.cuda.is_available() else "cpu"
     ckpt = torch.load(model_path, map_location=device)
     hp = ckpt.get("hparams", ckpt.get("hyper_parameters", {}))