From 3b53bd38c8efcc5ba54421fbfa90d047f1a61f82 Mon Sep 17 00:00:00 2001 From: itsainii Date: Tue, 17 Dec 2024 16:47:18 +0800 Subject: [PATCH] feat: Add Easyocr parameter recog_network (#613) * Update easyocr_model.py Added this line of code to get recog_network of easyocr parameter recog_network = self.options.recog_network Signed-off-by: itsainii * Update pipeline_options.py Added this line in EasyOcrOptions function recog_network: Optional[str] = 'standard' Signed-off-by: itsainii * Add Easyocr recog_network parameter Signed-off-by: itsainii --------- Signed-off-by: itsainii --- docling/datamodel/pipeline_options.py | 1 + docling/models/easyocr_model.py | 1 + 2 files changed, 2 insertions(+) diff --git a/docling/datamodel/pipeline_options.py b/docling/datamodel/pipeline_options.py index cf1689d..0f750db 100644 --- a/docling/datamodel/pipeline_options.py +++ b/docling/datamodel/pipeline_options.py @@ -140,6 +140,7 @@ class EasyOcrOptions(OcrOptions): use_gpu: Optional[bool] = None model_storage_directory: Optional[str] = None + recog_network: Optional[str] = "standard" download_enabled: bool = True model_config = ConfigDict( diff --git a/docling/models/easyocr_model.py b/docling/models/easyocr_model.py index 5de1409..1a8294b 100644 --- a/docling/models/easyocr_model.py +++ b/docling/models/easyocr_model.py @@ -66,6 +66,7 @@ class EasyOcrModel(BaseOcrModel): lang_list=self.options.lang, gpu=use_gpu, model_storage_directory=self.options.model_storage_directory, + recog_network=self.options.recog_network, download_enabled=self.options.download_enabled, verbose=False, )