docling-ibm-models 3.0.0__tar.gz → 3.1.1__tar.gz

{docling_ibm_models-3.0.0 → docling_ibm_models-3.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: docling-ibm-models
-Version: 3.0.0
+Version: 3.1.1
 Summary: This package contains the AI models used by the Docling PDF conversion package
 License: MIT
 Keywords: docling,convert,document,pdf,layout model,segmentation,table structure,table former
@@ -23,7 +23,8 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Dist: Pillow (>=10.0.0,<11.0.0)
 Requires-Dist: huggingface_hub (>=0.23,<1)
 Requires-Dist: jsonlines (>=3.1.0,<4.0.0)
-Requires-Dist: numpy (>=1.24.4,<3.0.0)
+Requires-Dist: numpy (>=1.24.4,<2.0.0) ; sys_platform == "darwin" and platform_machine == "x86_64"
+Requires-Dist: numpy (>=1.24.4,<3.0.0) ; sys_platform != "darwin" or platform_machine != "x86_64"
 Requires-Dist: opencv-python-headless (>=4.6.0.66,<5.0.0.0)
 Requires-Dist: safetensors[torch] (>=0.4.3,<1)
 Requires-Dist: torch (>=2.2.2,<3.0.0)

{docling_ibm_models-3.0.0 → docling_ibm_models-3.1.1}/docling_ibm_models/layoutmodel/layout_predictor.py

@@ -5,7 +5,7 @@
 import logging
 import os
 from collections.abc import Iterable
-from typing import Union
+from typing import Set, Union
 
 import numpy as np
 import torch
@@ -26,6 +26,8 @@ class LayoutPredictor:
         artifact_path: str,
         device: str = "cpu",
         num_threads: int = 4,
+        base_threshold: float = 0.3,
+        blacklist_classes: Set[str] = set(),
     ):
         """
         Provide the artifact path that contains the LayoutModel file
@@ -63,10 +65,10 @@ class LayoutPredictor:
         }
 
         # Blacklisted classes
-        self._black_classes = set()  # ["Form", "Key-Value Region"])
+        self._black_classes = blacklist_classes  # set(["Form", "Key-Value Region"])
 
         # Set basic params
-        self._threshold = 0.3  # Score threshold
+        self._threshold = base_threshold  # Score threshold
         self._image_size = 640
         self._size = np.asarray([[self._image_size, self._image_size]], dtype=np.int64)
 

{docling_ibm_models-3.0.0 → docling_ibm_models-3.1.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "docling-ibm-models"
-version = "3.0.0"  # DO NOT EDIT, updated automatically
+version = "3.1.1"  # DO NOT EDIT, updated automatically
 description = "This package contains the AI models used by the Docling PDF conversion package"
 authors = ["Nikos Livathinos <nli@zurich.ibm.com>", "Maxim Lysak <mly@zurich.ibm.com>", "Ahmed Nassar <ahn@zurich.ibm.com>", "Christoph Auer <cau@zurich.ibm.com>", "Michele Dolfi <dol@zurich.ibm.com>", "Peter Staar <taa@zurich.ibm.com>"]
 license = "MIT"
@@ -25,7 +25,10 @@ python = "^3.9"
 torch = "^2.2.2"
 torchvision = "^0"
 transformers = "^4.42.0"
-numpy = ">=1.24.4,<3.0.0"
+numpy = [
+    { version = ">=1.24.4,<3.0.0", markers = 'sys_platform != "darwin" or platform_machine != "x86_64"' },
+    { version = ">=1.24.4,<2.0.0", markers = 'sys_platform == "darwin" and platform_machine == "x86_64"' },
+]
 jsonlines = "^3.1.0"
 Pillow = "^10.0.0"
 tqdm = "^4.64.0"