From fc5b9bdf595d2ae8fee9961a7c02699a8d61b4c5 Mon Sep 17 00:00:00 2001 From: shamoon <4887959+shamoon@users.noreply.github.com> Date: Sun, 31 Aug 2025 15:41:33 -0700 Subject: [PATCH] Cache classifier loading with lru_cache --- src/documents/classifier.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/documents/classifier.py b/src/documents/classifier.py index 155f5b20c..02094fa74 100644 --- a/src/documents/classifier.py +++ b/src/documents/classifier.py @@ -4,6 +4,7 @@ import logging import pickle import re import warnings +from functools import lru_cache from hashlib import sha256 from pathlib import Path from typing import TYPE_CHECKING @@ -52,6 +53,7 @@ class ClassifierModelCorruptError(Exception): pass +@lru_cache(maxsize=1) def load_classifier(*, raise_exception: bool = False) -> DocumentClassifier | None: if not settings.MODEL_FILE.is_file(): logger.debug(