diff --git a/src/eynollah/eynollah.py b/src/eynollah/eynollah.py index c10b22e..68a9580 100644 --- a/src/eynollah/eynollah.py +++ b/src/eynollah/eynollah.py @@ -24,11 +24,12 @@ from ocrd import OcrdPage from ocrd_utils import getLogger import cv2 import numpy as np -from transformers import TrOCRProcessor -from PIL import Image import torch from difflib import SequenceMatcher as sq -from transformers import VisionEncoderDecoderModel +try: + from transformers import TrOCRProcessor, VisionEncoderDecoderModel +except ImportError: + TrOCRProcessor = VisionEncoderDecoderModel = None from numba import cuda import copy from scipy.signal import find_peaks