diff options
Diffstat (limited to 'src/text_recognizer/util.py')
| -rw-r--r-- | src/text_recognizer/util.py | 21 |
1 files changed, 11 insertions, 10 deletions
diff --git a/src/text_recognizer/util.py b/src/text_recognizer/util.py index 6c07c60..b431e22 100644 --- a/src/text_recognizer/util.py +++ b/src/text_recognizer/util.py @@ -21,20 +21,21 @@ def read_image(image_uri: Union[Path, str], grayscale: bool = False) -> np.ndarr return cv2.imdecode(image_array, imread_flag) else: raise ValueError( - "Url does not start with http, therfore not safe to open..." + "Url does not start with http, therefore not safe to open..." ) from None imread_flag = cv2.IMREAD_GRAYSCALE if grayscale else cv2.IMREAD_COLOR local_file = os.path.exists(image_uri) - try: - image = None - if local_file: - image = read_image_from_filename(image_uri, imread_flag) - else: - image = read_image_from_url(image_uri, imread_flag) - assert image is not None - except Exception as e: - raise ValueError(f"Could not load image at {image_uri}: {e}") + image = None + + if local_file: + image = read_image_from_filename(image_uri, imread_flag) + else: + image = read_image_from_url(image_uri, imread_flag) + + if image is None: + raise ValueError(f"Could not load image at {image_uri}") + return image |