diff --git a/antiabuse/antiporn/__init__.py b/antiabuse/antiporn/__init__.py index ec118a9..1ebf285 100644 --- a/antiabuse/antiporn/__init__.py +++ b/antiabuse/antiporn/__init__.py @@ -7,7 +7,7 @@ from io import BytesIO from pathlib import Path -_MODEL_PATH_BASE = Path(__file__).parent.parent.parent / 'antiporn' +_MODEL_PATH_BASE = Path(__file__).parent.parent / 'antiporn' _MODEL = bytearray() for file_path in sorted(_MODEL_PATH_BASE.glob('model.onnx.part*')): diff --git a/antiabuse/antispam/gibberishdetector/__init__.py b/antiabuse/antispam/gibberishdetector/__init__.py index bc2f204..93ef892 100644 --- a/antiabuse/antispam/gibberishdetector/__init__.py +++ b/antiabuse/antispam/gibberishdetector/__init__.py @@ -253,7 +253,4 @@ def train_model(): train_model() else: # Load bigram probabilities for use - - # TODO - # bigram_probs = load_bigram_probs() - bigram_probs = None + bigram_probs = load_bigram_probs()