Fix stuff I broke

duolicious · Jan 1, 2025 · ee409ed · ee409ed
1 parent 1cdaa4c
commit ee409ed
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git 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
@@ -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()