docling-ibm-models 3.0.0__py3-none-any.whl → 3.1.1__py3-none-any.whl

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.dist-info → docling_ibm_models-3.1.1.dist-info}/METADATA

@@ -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.dist-info → docling_ibm_models-3.1.1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-docling_ibm_models/layoutmodel/layout_predictor.py,sha256=aVLRsKSR_DBiCyTPCQipvkHdcmUYSPOvhp0KVuuVttM,5567
+docling_ibm_models/layoutmodel/layout_predictor.py,sha256=ArVgs7FBOiu23TC-JoybcaTp7F7a4BgYC8uRVxTgx4E,5681
 docling_ibm_models/tableformer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 docling_ibm_models/tableformer/common.py,sha256=2zgGZBFf4fXytEaXrZR2NU6FWdX2kxO0DHlGZmuvpNQ,3230
 docling_ibm_models/tableformer/data_management/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -22,7 +22,7 @@ docling_ibm_models/tableformer/utils/app_profiler.py,sha256=Pb7o1zcikKXh7ninaNt4
 docling_ibm_models/tableformer/utils/mem_monitor.py,sha256=ycZ07fUBVVKKLTVGF54jGPDM2aTkKuZWk1kMbOS0wwQ,6353
 docling_ibm_models/tableformer/utils/torch_utils.py,sha256=uN0rK9mSXy1ewBnBnILrWebJhhVU4N-XJZBqNiLJwlQ,8893
 docling_ibm_models/tableformer/utils/utils.py,sha256=8Bxf1rEn977lFbY9NX0r5xh9PvxIRipQZX_EZW92XfA,10980
-docling_ibm_models-3.0.0.dist-info/LICENSE,sha256=mBb7ErEcM8VS9OhiGHnQ2kk75HwPhr54W1Oiz3965MY,1088
-docling_ibm_models-3.0.0.dist-info/METADATA,sha256=Od5o3mMpPJnIh7jaWbI1iqnwayYizM53ALYrqEN5UHQ,7023
-docling_ibm_models-3.0.0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-docling_ibm_models-3.0.0.dist-info/RECORD,,
+docling_ibm_models-3.1.1.dist-info/LICENSE,sha256=mBb7ErEcM8VS9OhiGHnQ2kk75HwPhr54W1Oiz3965MY,1088
+docling_ibm_models-3.1.1.dist-info/METADATA,sha256=71pKGRH9yv4TvZChCCoyyfOUD1FO4biNy1L3nCVKbIU,7181
+docling_ibm_models-3.1.1.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+docling_ibm_models-3.1.1.dist-info/RECORD,,