onnxtr 0.7.1__tar.gz → 0.8.1__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: onnxtr
-Version: 0.7.1
+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
@@ -264,6 +264,7 @@ Requires-Dist: pytest>=5.3.2; extra == "testing"
 Requires-Dist: coverage[toml]>=4.5.4; extra == "testing"
 Requires-Dist: requests>=2.20.0; extra == "testing"
 Requires-Dist: pytest-memray>=1.7.0; extra == "testing"
+Requires-Dist: psutil>=7.0.0; extra == "testing"
 Provides-Extra: quality
 Requires-Dist: ruff>=0.1.5; extra == "quality"
 Requires-Dist: mypy>=0.812; extra == "quality"
@@ -278,6 +279,7 @@ Requires-Dist: pytest>=5.3.2; extra == "dev"
 Requires-Dist: coverage[toml]>=4.5.4; extra == "dev"
 Requires-Dist: requests>=2.20.0; extra == "dev"
 Requires-Dist: pytest-memray>=1.7.0; extra == "dev"
+Requires-Dist: psutil>=7.0.0; extra == "dev"
 Requires-Dist: ruff>=0.1.5; extra == "dev"
 Requires-Dist: mypy>=0.812; extra == "dev"
 Requires-Dist: pre-commit>=2.17.0; extra == "dev"
@@ -292,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.7.1?artifact_id=tar-gz)](https://socket.dev/pypi/package/onnxtr/overview/0.7.1/tar-gz)
-[![Pypi](https://img.shields.io/badge/pypi-v0.7.1-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)
@@ -323,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).
 
@@ -353,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:**
@@ -480,6 +483,32 @@ predictor = ocr_predictor(
 )
 ```
 
+You can also dynamically configure whether the memory arena should shrink:
+
+```python
+from random import random
+from onnxruntime import RunOptions, SessionOptions
+
+from onnxtr.models import ocr_predictor, EngineConfig
+
+def arena_shrinkage_handler(run_options: RunOptions) -> RunOptions:
+  """
+  Shrink the memory arena on 10% of inference runs.
+  """
+  if random() < 0.1:
+    run_options.add_run_config_entry("memory.enable_memory_arena_shrinkage", "cpu:0")
+  return run_options
+
+engine_config = EngineConfig(run_options_provider=arena_shrinkage_handler)
+engine_config.session_options.enable_mem_pattern = False
+
+predictor = ocr_predictor(
+    det_engine_cfg=engine_config,
+    reco_engine_cfg=engine_config,
+    clf_engine_cfg=engine_config
+)
+```
+
 </details>
 
 ## Loading custom exported models

{onnxtr-0.7.1 → 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.7.1?artifact_id=tar-gz)](https://socket.dev/pypi/package/onnxtr/overview/0.7.1/tar-gz)
-[![Pypi](https://img.shields.io/badge/pypi-v0.7.1-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:**
@@ -195,6 +196,32 @@ predictor = ocr_predictor(
 )
 ```
 
+You can also dynamically configure whether the memory arena should shrink:
+
+```python
+from random import random
+from onnxruntime import RunOptions, SessionOptions
+
+from onnxtr.models import ocr_predictor, EngineConfig
+
+def arena_shrinkage_handler(run_options: RunOptions) -> RunOptions:
+  """
+  Shrink the memory arena on 10% of inference runs.
+  """
+  if random() < 0.1:
+    run_options.add_run_config_entry("memory.enable_memory_arena_shrinkage", "cpu:0")
+  return run_options
+
+engine_config = EngineConfig(run_options_provider=arena_shrinkage_handler)
+engine_config.session_options.enable_mem_pattern = False
+
+predictor = ocr_predictor(
+    det_engine_cfg=engine_config,
+    reco_engine_cfg=engine_config,
+    clf_engine_cfg=engine_config
+)
+```
+
 </details>
 
 ## Loading custom exported models

{onnxtr-0.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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 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 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.
@@ -53,7 +53,7 @@ class DetectionPostProcessor(NestedObject):
 
         else:
             mask: np.ndarray = np.zeros((h, w), np.int32)
-            cv2.fillPoly(mask, [points.astype(np.int32)], 1.0)  # type: ignore[call-overload]
+            cv2.fillPoly(mask, [points.astype(np.int32)], 1.0)
             product = pred * mask
             return np.sum(product) / np.count_nonzero(product)
 

{onnxtr-0.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → onnxtr-0.8.1}/onnxtr/models/engine.py

@@ -1,17 +1,19 @@
-# 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 logging
 import os
-from typing import Any
+from collections.abc import Callable
+from typing import Any, TypeAlias
 
 import numpy as np
 from onnxruntime import (
     ExecutionMode,
     GraphOptimizationLevel,
     InferenceSession,
+    RunOptions,
     SessionOptions,
     get_available_providers,
     get_device,
@@ -23,7 +25,9 @@ set_default_logger_severity(int(os.getenv("ORT_LOG_SEVERITY_LEVEL", 4)))
 from onnxtr.utils.data import download_from_url
 from onnxtr.utils.geometry import shape_translate
 
-__all__ = ["EngineConfig"]
+__all__ = ["EngineConfig", "RunOptionsProvider"]
+
+RunOptionsProvider: TypeAlias = Callable[[RunOptions], RunOptions]
 
 
 class EngineConfig:
@@ -38,9 +42,11 @@ class EngineConfig:
         self,
         providers: list[tuple[str, dict[str, Any]]] | list[str] | None = None,
         session_options: SessionOptions | None = None,
+        run_options_provider: RunOptionsProvider | None = None,
     ):
         self._providers = providers or self._init_providers()
         self._session_options = session_options or self._init_sess_opts()
+        self.run_options_provider = run_options_provider
 
     def _init_providers(self) -> list[tuple[str, dict[str, Any]]]:
         providers: Any = [("CPUExecutionProvider", {"arena_extend_strategy": "kSameAsRequested"})]
@@ -59,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:
@@ -79,7 +87,7 @@ class EngineConfig:
         return self._session_options
 
     def __repr__(self) -> str:
-        return f"EngineConfig(providers={self.providers}"
+        return f"EngineConfig(providers={self.providers})"
 
 
 class Engine:
@@ -100,6 +108,7 @@ class Engine:
         self.model_path = archive_path
         self.session_options = engine_cfg.session_options
         self.providers = engine_cfg.providers
+        self.run_options_provider = engine_cfg.run_options_provider
         self.runtime = InferenceSession(archive_path, providers=self.providers, sess_options=self.session_options)
         self.runtime_inputs = self.runtime.get_inputs()[0]
         self.tf_exported = int(self.runtime_inputs.shape[-1]) == 3
@@ -109,6 +118,9 @@ class Engine:
         self.output_name = [output.name for output in self.runtime.get_outputs()]
 
     def run(self, inputs: np.ndarray) -> np.ndarray:
+        run_options = RunOptions()
+        if self.run_options_provider is not None:
+            run_options = self.run_options_provider(run_options)
         if self.tf_exported:
             inputs = shape_translate(inputs, format="BHWC")  # sanity check
         else:
@@ -117,8 +129,12 @@ class Engine:
             inputs = np.broadcast_to(inputs, (self.fixed_batch_size, *inputs.shape))
             # combine the results
             logits = np.concatenate(
-                [self.runtime.run(self.output_name, {self.runtime_inputs.name: batch})[0] for batch in inputs], axis=0
+                [
+                    self.runtime.run(self.output_name, {self.runtime_inputs.name: batch}, run_options=run_options)[0]
+                    for batch in inputs
+                ],
+                axis=0,
             )
         else:
-            logits = self.runtime.run(self.output_name, {self.runtime_inputs.name: inputs})[0]
+            logits = self.runtime.run(self.output_name, {self.runtime_inputs.name: inputs}, run_options=run_options)[0]
         return shape_translate(logits, format="BHWC")

{onnxtr-0.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.
@@ -22,9 +22,9 @@ default_cfgs: dict[str, dict[str, Any]] = {
         "mean": (0.694, 0.695, 0.693),
         "std": (0.299, 0.296, 0.301),
         "input_shape": (3, 32, 128),
-        "vocab": VOCABS["legacy_french"],
-        "url": "https://github.com/felixdittrich92/OnnxTR/releases/download/v0.0.1/crnn_vgg16_bn-662979cc.onnx",
-        "url_8_bit": "https://github.com/felixdittrich92/OnnxTR/releases/download/v0.1.2/crnn_vgg16_bn_static_8_bit-bce050c7.onnx",
+        "vocab": VOCABS["french"],
+        "url": "https://github.com/felixdittrich92/OnnxTR/releases/download/v0.7.1/crnn_vgg16_bn-743599aa.onnx",
+        "url_8_bit": "https://github.com/felixdittrich92/OnnxTR/releases/download/v0.7.1/crnn_vgg16_bn_static_8_bit-df1b594d.onnx",
     },
     "crnn_mobilenet_v3_small": {
         "mean": (0.694, 0.695, 0.693),

{onnxtr-0.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.7.1 → 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.
@@ -264,8 +264,6 @@ VOCABS["estonian"] = VOCABS["english"] + "šžõäöüŠŽÕÄÖÜ"
 VOCABS["esperanto"] = re.sub(r"[QqWwXxYy]", "", VOCABS["english"]) + "ĉĝĥĵŝŭĈĜĤĴŜŬ" + "₷"
 
 VOCABS["french"] = VOCABS["english"] + "àâéèêëîïôùûüçÀÂÉÈÊËÎÏÔÙÛÜÇ"
-# NOTE: legacy french is outdated, but kept for compatibility
-VOCABS["legacy_french"] = VOCABS["latin"] + "°" + "àâéèêëîïôùûçÀÂÉÈËÎÏÔÙÛÇ" + _BASE_VOCABS["currency"]
 
 VOCABS["finnish"] = VOCABS["english"] + "äöÄÖ"
 

onnxtr-0.8.1/onnxtr/version.py

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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: onnxtr
-Version: 0.7.1
+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
@@ -264,6 +264,7 @@ Requires-Dist: pytest>=5.3.2; extra == "testing"
 Requires-Dist: coverage[toml]>=4.5.4; extra == "testing"
 Requires-Dist: requests>=2.20.0; extra == "testing"
 Requires-Dist: pytest-memray>=1.7.0; extra == "testing"
+Requires-Dist: psutil>=7.0.0; extra == "testing"
 Provides-Extra: quality
 Requires-Dist: ruff>=0.1.5; extra == "quality"
 Requires-Dist: mypy>=0.812; extra == "quality"
@@ -278,6 +279,7 @@ Requires-Dist: pytest>=5.3.2; extra == "dev"
 Requires-Dist: coverage[toml]>=4.5.4; extra == "dev"
 Requires-Dist: requests>=2.20.0; extra == "dev"
 Requires-Dist: pytest-memray>=1.7.0; extra == "dev"
+Requires-Dist: psutil>=7.0.0; extra == "dev"
 Requires-Dist: ruff>=0.1.5; extra == "dev"
 Requires-Dist: mypy>=0.812; extra == "dev"
 Requires-Dist: pre-commit>=2.17.0; extra == "dev"
@@ -292,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.7.1?artifact_id=tar-gz)](https://socket.dev/pypi/package/onnxtr/overview/0.7.1/tar-gz)
-[![Pypi](https://img.shields.io/badge/pypi-v0.7.1-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)
@@ -323,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).
 
@@ -353,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:**
@@ -480,6 +483,32 @@ predictor = ocr_predictor(
 )
 ```
 
+You can also dynamically configure whether the memory arena should shrink:
+
+```python
+from random import random
+from onnxruntime import RunOptions, SessionOptions
+
+from onnxtr.models import ocr_predictor, EngineConfig
+
+def arena_shrinkage_handler(run_options: RunOptions) -> RunOptions:
+  """
+  Shrink the memory arena on 10% of inference runs.
+  """
+  if random() < 0.1:
+    run_options.add_run_config_entry("memory.enable_memory_arena_shrinkage", "cpu:0")
+  return run_options
+
+engine_config = EngineConfig(run_options_provider=arena_shrinkage_handler)
+engine_config.session_options.enable_mem_pattern = False
+
+predictor = ocr_predictor(
+    det_engine_cfg=engine_config,
+    reco_engine_cfg=engine_config,
+    clf_engine_cfg=engine_config
+)
+```
+
 </details>
 
 ## Loading custom exported models

{onnxtr-0.7.1 → 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
@@ -28,6 +28,7 @@ pytest>=5.3.2
 coverage[toml]>=4.5.4
 requests>=2.20.0
 pytest-memray>=1.7.0
+psutil>=7.0.0
 ruff>=0.1.5
 mypy>=0.812
 pre-commit>=2.17.0
@@ -61,6 +62,7 @@ pytest>=5.3.2
 coverage[toml]>=4.5.4
 requests>=2.20.0
 pytest-memray>=1.7.0
+psutil>=7.0.0
 
 [viz]
 matplotlib>=3.1.0

{onnxtr-0.7.1 → 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",
@@ -81,6 +81,7 @@ testing = [
     "coverage[toml]>=4.5.4",
     "requests>=2.20.0",
     "pytest-memray>=1.7.0",
+    "psutil>=7.0.0",
 ]
 quality = [
     "ruff>=0.1.5",
@@ -101,6 +102,7 @@ dev = [
     "coverage[toml]>=4.5.4",
     "requests>=2.20.0",
     "pytest-memray>=1.7.0",
+    "psutil>=7.0.0",
     # Quality
     "ruff>=0.1.5",
     "mypy>=0.812",

{onnxtr-0.7.1 → 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.7.1a0")
+VERSION = os.getenv("BUILD_VERSION", "0.8.1a0")
 
 
 if __name__ == "__main__":

onnxtr-0.7.1/onnxtr/version.py

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