diff --git a/qurator/sbb_ner/webapp/app.py b/qurator/sbb_ner/webapp/app.py index 4a23ca0..8820a69 100644 --- a/qurator/sbb_ner/webapp/app.py +++ b/qurator/sbb_ner/webapp/app.py @@ -244,6 +244,9 @@ def ner(model_id=None): if not token.startswith('##') and word_pred == 'X': word_pred = 'O' + if word_pred == '[SEP]': + word_pred = 'O' + token = token[2:] if token.startswith('##') else token word += token