From 7211086cb24f156c5e9d6fb60bbf72fcd632a9d3 Mon Sep 17 00:00:00 2001 From: CoderKang Date: Tue, 9 Sep 2025 23:12:53 +0800 Subject: [PATCH] fix(ocr): switch RapidOCR lang type from LangDet to LangRec --- machine-learning/immich_ml/models/ocr/recognition.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/machine-learning/immich_ml/models/ocr/recognition.py b/machine-learning/immich_ml/models/ocr/recognition.py index f88bd9ec3c..0f1394304c 100644 --- a/machine-learning/immich_ml/models/ocr/recognition.py +++ b/machine-learning/immich_ml/models/ocr/recognition.py @@ -7,7 +7,7 @@ from rapidocr.ch_ppocr_rec import TextRecInput from rapidocr.ch_ppocr_rec import TextRecognizer as RapidTextRecognizer from rapidocr.inference_engine.base import FileInfo, InferSession from rapidocr.utils import DownloadFile, DownloadFileInput -from rapidocr.utils.typings import EngineType, LangDet, OCRVersion, TaskType +from rapidocr.utils.typings import EngineType, LangRec, OCRVersion, TaskType from rapidocr.utils.typings import ModelType as RapidModelType from immich_ml.config import log, settings @@ -38,7 +38,7 @@ class TextRecognizer(InferenceModel): engine_type=EngineType.ONNXRUNTIME, ocr_version=OCRVersion.PPOCRV5, task_type=TaskType.REC, - lang_type=LangDet.CH, + lang_type=LangRec.CH, model_type=RapidModelType.MOBILE if "mobile" in self.model_name else RapidModelType.SERVER, ) )