diff --git a/docling/models/base_ocr_model.py b/docling/models/base_ocr_model.py index 46e5ef7a..dfad8fd2 100644 --- a/docling/models/base_ocr_model.py +++ b/docling/models/base_ocr_model.py @@ -3,7 +3,7 @@ import logging from abc import abstractmethod from collections.abc import Iterable from pathlib import Path -from typing import List, Optional, Type +from typing import TYPE_CHECKING, List, Optional, Type import numpy as np from docling_core.types.doc import BoundingBox, CoordOrigin @@ -20,6 +20,9 @@ from docling.models.base_model import BaseModelWithOptions, BasePageModel _log = logging.getLogger(__name__) +if TYPE_CHECKING: + from scipy.ndimage import binary_dilation, find_objects, label + class BaseOcrModel(BasePageModel, BaseModelWithOptions): def __init__( diff --git a/docling/models/plugins/defaults.py b/docling/models/plugins/defaults.py index 3fcb8b5f..87798478 100644 --- a/docling/models/plugins/defaults.py +++ b/docling/models/plugins/defaults.py @@ -1,5 +1,3 @@ - - def ocr_engines(): from docling.models.easyocr_model import EasyOcrModel from docling.models.ocr_mac_model import OcrMacModel diff --git a/docling/utils/accelerator_utils.py b/docling/utils/accelerator_utils.py index d7551827..826b3657 100644 --- a/docling/utils/accelerator_utils.py +++ b/docling/utils/accelerator_utils.py @@ -17,6 +17,7 @@ def decide_device( 2. User-defined: Check if the device actually exists, otherwise fall-back to CPU """ import torch + device = "cpu" has_cuda = torch.backends.cuda.is_built() and torch.cuda.is_available()