onnxtr 0.8.0__tar.gz → 0.8.1__tar.gz

{onnxtr-0.8.0 → onnxtr-0.8.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: onnxtr
-Version: 0.8.0
+Version: 0.8.1
 Summary: Onnx Text Recognition (OnnxTR): docTR Onnx-Wrapper for high-performance OCR on documents.
 Author-email: Felix Dittrich <felixdittrich92@gmail.com>
 Maintainer: Felix Dittrich
@@ -227,11 +227,11 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: numpy<3.0.0,>=1.16.0
 Requires-Dist: scipy<2.0.0,>=1.4.0
-Requires-Dist: pypdfium2<5.0.0,>=4.11.0
+Requires-Dist: pypdfium2<6.0.0,>=4.11.0
 Requires-Dist: pyclipper<2.0.0,>=1.2.0
 Requires-Dist: rapidfuzz<4.0.0,>=3.0.0
 Requires-Dist: langdetect<2.0.0,>=1.0.9
-Requires-Dist: huggingface-hub<1.0.0,>=0.23.0
+Requires-Dist: huggingface-hub<2.0.0,>=0.23.0
 Requires-Dist: Pillow>=9.2.0
 Requires-Dist: defusedxml>=0.7.0
 Requires-Dist: anyascii>=0.3.2
@@ -294,8 +294,8 @@ Dynamic: license-file
 [![codecov](https://codecov.io/gh/felixdittrich92/OnnxTR/graph/badge.svg?token=WVFRCQBOLI)](https://codecov.io/gh/felixdittrich92/OnnxTR)
 [![Codacy Badge](https://app.codacy.com/project/badge/Grade/4fff4d764bb14fb8b4f4afeb9587231b)](https://app.codacy.com/gh/felixdittrich92/OnnxTR/dashboard?utm_source=gh&utm_medium=referral&utm_content=&utm_campaign=Badge_grade)
 [![CodeFactor](https://www.codefactor.io/repository/github/felixdittrich92/onnxtr/badge)](https://www.codefactor.io/repository/github/felixdittrich92/onnxtr)
-[![Socket Badge](https://socket.dev/api/badge/pypi/package/onnxtr/0.8.0?artifact_id=tar-gz)](https://socket.dev/pypi/package/onnxtr/overview/0.8.0/tar-gz)
-[![Pypi](https://img.shields.io/badge/pypi-v0.8.0-blue.svg)](https://pypi.org/project/OnnxTR/)
+[![Socket Badge](https://socket.dev/api/badge/pypi/package/onnxtr/0.8.1?artifact_id=tar-gz)](https://socket.dev/pypi/package/onnxtr/overview/0.8.1/tar-gz)
+[![Pypi](https://img.shields.io/badge/pypi-v0.8.1-blue.svg)](https://pypi.org/project/OnnxTR/)
 [![Docker Images](https://img.shields.io/badge/Docker-4287f5?style=flat&logo=docker&logoColor=white)](https://github.com/felixdittrich92/OnnxTR/pkgs/container/onnxtr)
 [![Hugging Face Spaces](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-blue)](https://huggingface.co/spaces/Felix92/OnnxTR-OCR)
 ![PyPI - Downloads](https://img.shields.io/pypi/dm/onnxtr)
@@ -325,7 +325,7 @@ You can then install the latest release of the package using [pypi](https://pypi
 
 **NOTE:**
 
-Currently supported execution providers by default are: CPU, CUDA (NVIDIA GPU), OpenVINO (Intel CPU | GPU).
+Currently supported execution providers by default are: CPU, CUDA (NVIDIA GPU), OpenVINO (Intel CPU | GPU), CoreML (Apple Silicon).
 
 For GPU support please take a look at: [ONNX Runtime](https://onnxruntime.ai/getting-started).
 
@@ -355,6 +355,7 @@ If you have:
 
 - a NVIDIA GPU, use one of the `gpu` variants
 - an Intel CPU or GPU, use one of the `openvino` variants
+- an Apple Silicon Mac, use one of the `cpu` variants (CoreML is auto-detected)
 - otherwise, use one of the `cpu` variants
 
 **OpenVINO:**

{onnxtr-0.8.0 → onnxtr-0.8.1}/README.md

@@ -7,8 +7,8 @@
 [![codecov](https://codecov.io/gh/felixdittrich92/OnnxTR/graph/badge.svg?token=WVFRCQBOLI)](https://codecov.io/gh/felixdittrich92/OnnxTR)
 [![Codacy Badge](https://app.codacy.com/project/badge/Grade/4fff4d764bb14fb8b4f4afeb9587231b)](https://app.codacy.com/gh/felixdittrich92/OnnxTR/dashboard?utm_source=gh&utm_medium=referral&utm_content=&utm_campaign=Badge_grade)
 [![CodeFactor](https://www.codefactor.io/repository/github/felixdittrich92/onnxtr/badge)](https://www.codefactor.io/repository/github/felixdittrich92/onnxtr)
-[![Socket Badge](https://socket.dev/api/badge/pypi/package/onnxtr/0.8.0?artifact_id=tar-gz)](https://socket.dev/pypi/package/onnxtr/overview/0.8.0/tar-gz)
-[![Pypi](https://img.shields.io/badge/pypi-v0.8.0-blue.svg)](https://pypi.org/project/OnnxTR/)
+[![Socket Badge](https://socket.dev/api/badge/pypi/package/onnxtr/0.8.1?artifact_id=tar-gz)](https://socket.dev/pypi/package/onnxtr/overview/0.8.1/tar-gz)
+[![Pypi](https://img.shields.io/badge/pypi-v0.8.1-blue.svg)](https://pypi.org/project/OnnxTR/)
 [![Docker Images](https://img.shields.io/badge/Docker-4287f5?style=flat&logo=docker&logoColor=white)](https://github.com/felixdittrich92/OnnxTR/pkgs/container/onnxtr)
 [![Hugging Face Spaces](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-blue)](https://huggingface.co/spaces/Felix92/OnnxTR-OCR)
 ![PyPI - Downloads](https://img.shields.io/pypi/dm/onnxtr)
@@ -38,7 +38,7 @@ You can then install the latest release of the package using [pypi](https://pypi
 
 **NOTE:**
 
-Currently supported execution providers by default are: CPU, CUDA (NVIDIA GPU), OpenVINO (Intel CPU | GPU).
+Currently supported execution providers by default are: CPU, CUDA (NVIDIA GPU), OpenVINO (Intel CPU | GPU), CoreML (Apple Silicon).
 
 For GPU support please take a look at: [ONNX Runtime](https://onnxruntime.ai/getting-started).
 
@@ -68,6 +68,7 @@ If you have:
 
 - a NVIDIA GPU, use one of the `gpu` variants
 - an Intel CPU or GPU, use one of the `openvino` variants
+- an Apple Silicon Mac, use one of the `cpu` variants (CoreML is auto-detected)
 - otherwise, use one of the `cpu` variants
 
 **OpenVINO:**

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/contrib/artefacts.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/contrib/base.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/file_utils.py

@@ -1,10 +1,9 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.
 
 import importlib.metadata
-import importlib.util
 import logging
 
 __all__ = ["requires_package"]

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/io/elements.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/io/html.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/io/image.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/io/pdf.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/io/reader.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/_utils.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.
@@ -25,6 +25,8 @@ def get_max_width_length_ratio(contour: np.ndarray) -> float:
         the maximum shape ratio
     """
     _, (w, h), _ = cv2.minAreaRect(contour)
+    if w == 0 or h == 0:
+        return 0.0
     return max(w / h, h / w)
 
 
@@ -52,7 +54,7 @@ def estimate_orientation(
         the estimated angle of the page (clockwise, negative for left side rotation, positive for right side rotation)
     """
     assert len(img.shape) == 3 and img.shape[-1] in [1, 3], f"Image shape {img.shape} not supported"
-    thresh = None
+
     # Convert image to grayscale if necessary
     if img.shape[-1] == 3:
         gray_img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
@@ -61,11 +63,14 @@ def estimate_orientation(
     else:
         thresh = img.astype(np.uint8)
 
-    page_orientation, orientation_confidence = general_page_orientation or (None, 0.0)
-    if page_orientation is not None and orientation_confidence >= min_confidence:
+    page_orientation, orientation_confidence = general_page_orientation or (0, 0.0)
+    is_confident = page_orientation is not None and orientation_confidence >= min_confidence
+    base_angle = page_orientation if is_confident else 0
+
+    if is_confident:
         # We rotate the image to the general orientation which improves the detection
         # No expand needed bitmap is already padded
-        thresh = rotate_image(thresh, -page_orientation)
+        thresh = rotate_image(thresh, -base_angle)
     else:  # That's only required if we do not work on the detection models bin map
         # try to merge words in lines
         (h, w) = img.shape[:2]
@@ -87,30 +92,63 @@ def estimate_orientation(
     angles = []
     for contour in contours[:n_ct]:
         _, (w, h), angle = cv2.minAreaRect(contour)
-        if w / h > ratio_threshold_for_lines:  # select only contours with ratio like lines
-            angles.append(angle)
-        elif w / h < 1 / ratio_threshold_for_lines:  # if lines are vertical, substract 90 degree
-            angles.append(angle - 90)
+
+        # OpenCV version-proof normalization: force 'w' to be the long side
+        # so the angle is consistently relative to the major axis.
+        # https://github.com/opencv/opencv/pull/28051/changes
+        if w < h:
+            w, h = h, w
+            angle -= 90
+
+        # Normalize angle to be within [-90, 90]
+        while angle <= -90:
+            angle += 180
+        while angle > 90:
+            angle -= 180
+
+        if h > 0:
+            if w / h > ratio_threshold_for_lines:  # select only contours with ratio like lines
+                angles.append(angle)
+            elif w / h < 1 / ratio_threshold_for_lines:  # if lines are vertical, substract 90 degree
+                angles.append(angle - 90)
 
     if len(angles) == 0:
-        estimated_angle = 0  # in case no angles is found
+        skew_angle = 0  # in case no angles is found
     else:
+        # median_low picks a value from the data to avoid outliers
         median = -median_low(angles)
-        estimated_angle = -round(median) if abs(median) != 0 else 0
+        skew_angle = -round(median) if abs(median) != 0 else 0
+
+        # Resolve the 90-degree flip ambiguity.
+        # If the estimation is exactly 90/-90, it's usually a vertical detection of horizontal lines.
+        if abs(skew_angle) == 90:
+            skew_angle = 0
 
     # combine with the general orientation and the estimated angle
-    if page_orientation is not None and orientation_confidence >= min_confidence:
+    # Apply the detected skew to our base orientation
+    final_angle = base_angle + skew_angle
+
+    # Standardize result to [-179, 180] range to handle wrap-around cases (e.g., 180 + -31)
+    while final_angle > 180:
+        final_angle -= 360
+    while final_angle <= -180:
+        final_angle += 360
+
+    if is_confident:
+        # If the estimated angle is perpendicular, treat it as 0 to avoid wrong flips
+        if abs(skew_angle) % 90 == 0:
+            return page_orientation
+
         # special case where the estimated angle is mostly wrong:
         # case 1: - and + swapped
         # case 2: estimated angle is completely wrong
         # so in this case we prefer the general page orientation
-        if abs(estimated_angle) == abs(page_orientation):
+        if abs(skew_angle) == abs(page_orientation) and page_orientation != 0:
             return page_orientation
-        estimated_angle = estimated_angle if page_orientation == 0 else page_orientation + estimated_angle
-        if estimated_angle > 180:
-            estimated_angle -= 360
 
-    return estimated_angle  # return the clockwise angle (negative - left side rotation, positive - right side rotation)
+    return int(
+        final_angle
+    )  # return the clockwise angle (negative - left side rotation, positive - right side rotation)
 
 
 def rectify_crops(

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/builder.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/classification/models/mobilenet.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/classification/predictor/base.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/classification/zoo.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/detection/_utils/base.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/detection/core.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/detection/models/differentiable_binarization.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/detection/models/fast.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/detection/models/linknet.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/detection/postprocessor/base.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/detection/predictor/base.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/detection/zoo.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/engine.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.
@@ -65,6 +65,8 @@ class EngineConfig:
                     },
                 ),
             )
+        elif "CoreMLExecutionProvider" in available_providers:  # pragma: no cover
+            providers.insert(0, ("CoreMLExecutionProvider", {}))
         return providers
 
     def _init_sess_opts(self) -> SessionOptions:

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/factory/hub.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.
@@ -7,18 +7,16 @@
 
 import json
 import logging
-import os
 import shutil
 import subprocess
+import tempfile
 import textwrap
 from pathlib import Path
 from typing import Any
 
 from huggingface_hub import (
     HfApi,
-    Repository,
     get_token,
-    get_token_permission,
     hf_hub_download,
     login,
 )
@@ -39,9 +37,9 @@ AVAILABLE_ARCHS = {
 def login_to_hub() -> None:  # pragma: no cover
     """Login to huggingface hub"""
     access_token = get_token()
-    if access_token is not None and get_token_permission(access_token):
+    if access_token is not None:
         logging.info("Huggingface Hub token found and valid")
-        login(token=access_token, write_permission=True)
+        login(token=access_token)
     else:
         login()
     # check if git lfs is installed
@@ -165,16 +163,23 @@ def push_to_hf_hub(
 
     commit_message = f"Add {model_name} model"
 
-    local_cache_dir = os.path.join(os.path.expanduser("~"), ".cache", "huggingface", "hub", model_name)
-    repo_url = HfApi().create_repo(model_name, token=get_token(), exist_ok=override)
-    repo = Repository(local_dir=local_cache_dir, clone_from=repo_url)
+    # Create repository
+    api = HfApi()
+    api.create_repo(model_name, token=get_token(), exist_ok=False)
 
-    with repo.commit(commit_message):
-        _save_model_and_config_for_hf_hub(model, repo.local_dir, arch=arch, task=task)
-        readme_path = Path(repo.local_dir) / "README.md"
+    # Save model files to a temporary directory
+    with tempfile.TemporaryDirectory() as tmp_dir:
+        _save_model_and_config_for_hf_hub(model, tmp_dir, arch=arch, task=task)
+        readme_path = Path(tmp_dir) / "README.md"
         readme_path.write_text(readme)
 
-    repo.git_push()
+        # Upload all files to the hub
+        api.upload_folder(
+            folder_path=tmp_dir,
+            repo_id=model_name,
+            commit_message=commit_message,
+            token=get_token(),
+        )
 
 
 def from_hub(repo_id: str, engine_cfg: EngineConfig | None = None, **kwargs: Any):

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/predictor/base.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/predictor/predictor.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/preprocessor/base.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/recognition/core.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/recognition/models/crnn.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/recognition/models/master.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/recognition/models/parseq.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/recognition/models/sar.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/recognition/models/viptr.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/recognition/models/vitstr.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/recognition/predictor/_utils.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/recognition/predictor/base.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/recognition/utils.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/recognition/zoo.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/models/zoo.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/transforms/base.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/utils/common_types.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/utils/data.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.
@@ -9,7 +9,6 @@ import hashlib
 import logging
 import os
 import re
-import urllib
 import urllib.error
 import urllib.request
 from pathlib import Path

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/utils/fonts.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/utils/geometry.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/utils/multithreading.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/utils/reconstitution.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/utils/repr.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/utils/visualization.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr/utils/vocabs.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr-0.8.1/onnxtr/version.py

@@ -0,0 +1 @@
+__version__ = 'v0.8.1'

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: onnxtr
-Version: 0.8.0
+Version: 0.8.1
 Summary: Onnx Text Recognition (OnnxTR): docTR Onnx-Wrapper for high-performance OCR on documents.
 Author-email: Felix Dittrich <felixdittrich92@gmail.com>
 Maintainer: Felix Dittrich
@@ -227,11 +227,11 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: numpy<3.0.0,>=1.16.0
 Requires-Dist: scipy<2.0.0,>=1.4.0
-Requires-Dist: pypdfium2<5.0.0,>=4.11.0
+Requires-Dist: pypdfium2<6.0.0,>=4.11.0
 Requires-Dist: pyclipper<2.0.0,>=1.2.0
 Requires-Dist: rapidfuzz<4.0.0,>=3.0.0
 Requires-Dist: langdetect<2.0.0,>=1.0.9
-Requires-Dist: huggingface-hub<1.0.0,>=0.23.0
+Requires-Dist: huggingface-hub<2.0.0,>=0.23.0
 Requires-Dist: Pillow>=9.2.0
 Requires-Dist: defusedxml>=0.7.0
 Requires-Dist: anyascii>=0.3.2
@@ -294,8 +294,8 @@ Dynamic: license-file
 [![codecov](https://codecov.io/gh/felixdittrich92/OnnxTR/graph/badge.svg?token=WVFRCQBOLI)](https://codecov.io/gh/felixdittrich92/OnnxTR)
 [![Codacy Badge](https://app.codacy.com/project/badge/Grade/4fff4d764bb14fb8b4f4afeb9587231b)](https://app.codacy.com/gh/felixdittrich92/OnnxTR/dashboard?utm_source=gh&utm_medium=referral&utm_content=&utm_campaign=Badge_grade)
 [![CodeFactor](https://www.codefactor.io/repository/github/felixdittrich92/onnxtr/badge)](https://www.codefactor.io/repository/github/felixdittrich92/onnxtr)
-[![Socket Badge](https://socket.dev/api/badge/pypi/package/onnxtr/0.8.0?artifact_id=tar-gz)](https://socket.dev/pypi/package/onnxtr/overview/0.8.0/tar-gz)
-[![Pypi](https://img.shields.io/badge/pypi-v0.8.0-blue.svg)](https://pypi.org/project/OnnxTR/)
+[![Socket Badge](https://socket.dev/api/badge/pypi/package/onnxtr/0.8.1?artifact_id=tar-gz)](https://socket.dev/pypi/package/onnxtr/overview/0.8.1/tar-gz)
+[![Pypi](https://img.shields.io/badge/pypi-v0.8.1-blue.svg)](https://pypi.org/project/OnnxTR/)
 [![Docker Images](https://img.shields.io/badge/Docker-4287f5?style=flat&logo=docker&logoColor=white)](https://github.com/felixdittrich92/OnnxTR/pkgs/container/onnxtr)
 [![Hugging Face Spaces](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-blue)](https://huggingface.co/spaces/Felix92/OnnxTR-OCR)
 ![PyPI - Downloads](https://img.shields.io/pypi/dm/onnxtr)
@@ -325,7 +325,7 @@ You can then install the latest release of the package using [pypi](https://pypi
 
 **NOTE:**
 
-Currently supported execution providers by default are: CPU, CUDA (NVIDIA GPU), OpenVINO (Intel CPU | GPU).
+Currently supported execution providers by default are: CPU, CUDA (NVIDIA GPU), OpenVINO (Intel CPU | GPU), CoreML (Apple Silicon).
 
 For GPU support please take a look at: [ONNX Runtime](https://onnxruntime.ai/getting-started).
 
@@ -355,6 +355,7 @@ If you have:
 
 - a NVIDIA GPU, use one of the `gpu` variants
 - an Intel CPU or GPU, use one of the `openvino` variants
+- an Apple Silicon Mac, use one of the `cpu` variants (CoreML is auto-detected)
 - otherwise, use one of the `cpu` variants
 
 **OpenVINO:**

{onnxtr-0.8.0 → onnxtr-0.8.1}/onnxtr.egg-info/requires.txt

@@ -1,10 +1,10 @@
 numpy<3.0.0,>=1.16.0
 scipy<2.0.0,>=1.4.0
-pypdfium2<5.0.0,>=4.11.0
+pypdfium2<6.0.0,>=4.11.0
 pyclipper<2.0.0,>=1.2.0
 rapidfuzz<4.0.0,>=3.0.0
 langdetect<2.0.0,>=1.0.9
-huggingface-hub<1.0.0,>=0.23.0
+huggingface-hub<2.0.0,>=0.23.0
 Pillow>=9.2.0
 defusedxml>=0.7.0
 anyascii>=0.3.2

{onnxtr-0.8.0 → onnxtr-0.8.1}/pyproject.toml

@@ -33,11 +33,11 @@ dependencies = [
     # Additional typing support is brought by numpy>=1.22.4, but core build sticks to >=1.16.0
     "numpy>=1.16.0,<3.0.0",
     "scipy>=1.4.0,<2.0.0",
-    "pypdfium2>=4.11.0,<5.0.0",
+    "pypdfium2>=4.11.0,<6.0.0",
     "pyclipper>=1.2.0,<2.0.0",
     "rapidfuzz>=3.0.0,<4.0.0",
     "langdetect>=1.0.9,<2.0.0",
-    "huggingface-hub>=0.23.0,<1.0.0",
+    "huggingface-hub>=0.23.0,<2.0.0",
     "Pillow>=9.2.0",
     "defusedxml>=0.7.0",
     "anyascii>=0.3.2",

{onnxtr-0.8.0 → onnxtr-0.8.1}/setup.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2026, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.
@@ -9,7 +9,7 @@ from pathlib import Path
 from setuptools import setup
 
 PKG_NAME = "onnxtr"
-VERSION = os.getenv("BUILD_VERSION", "0.8.0a0")
+VERSION = os.getenv("BUILD_VERSION", "0.8.1a0")
 
 
 if __name__ == "__main__":

onnxtr-0.8.0/onnxtr/version.py

@@ -1 +0,0 @@
-__version__ = 'v0.8.0'