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