From 81d176cd3d0d1d20c06d755778e7c7958efdba70 Mon Sep 17 00:00:00 2001 From: Michele Dolfi Date: Thu, 3 Oct 2024 13:38:01 +0200 Subject: [PATCH] add message for failed easyocr import Signed-off-by: Michele Dolfi --- docling/models/easyocr_model.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/docling/models/easyocr_model.py b/docling/models/easyocr_model.py index 5fd36ca8..13455b6b 100644 --- a/docling/models/easyocr_model.py +++ b/docling/models/easyocr_model.py @@ -18,7 +18,12 @@ class EasyOcrModel(BaseOcrModel): self.scale = 3 # multiplier for 72 dpi == 216 dpi. if self.enabled: - import easyocr + try: + import easyocr + except ImportError: + raise ImportError( + "EasyOCR is not installed. Please install it via `pip install easyocr` to use this OCR engine." + ) self.reader = easyocr.Reader(lang_list=self.options.lang)