mirror of
https://github.com/paperless-ngx/paperless-ngx.git
synced 2025-09-01 01:46:16 +00:00
Cache classifier loading with lru_cache
This commit is contained in:
@@ -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(
|
||||
|
Reference in New Issue
Block a user