forgot to sanitize mentions in output

This commit is contained in:
io 2021-08-17 05:55:29 +00:00
parent 97a2f5de00
commit 5db218e362
2 changed files with 7 additions and 2 deletions

View file

@ -54,7 +54,7 @@ def load_config(cfg_path):
return cfg
def remove_mention(cfg, sentence):
def remove_mentions(cfg, sentence):
# optionally remove mentions
if cfg['mention_handling'] == 1:
return re.sub(r"^\S*@\u200B\S*\s?", "", sentence)