docling-ibm-models 1.2.1__py3-none-any.whl → 1.3.1__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {docling_ibm_models-1.2.1.dist-info → docling_ibm_models-1.3.1.dist-info}/METADATA +6 -8
- {docling_ibm_models-1.2.1.dist-info → docling_ibm_models-1.3.1.dist-info}/RECORD +4 -4
- {docling_ibm_models-1.2.1.dist-info → docling_ibm_models-1.3.1.dist-info}/LICENSE +0 -0
- {docling_ibm_models-1.2.1.dist-info → docling_ibm_models-1.3.1.dist-info}/WHEEL +0 -0
@@ -1,6 +1,6 @@
|
|
1
1
|
Metadata-Version: 2.1
|
2
2
|
Name: docling-ibm-models
|
3
|
-
Version: 1.
|
3
|
+
Version: 1.3.1
|
4
4
|
Summary: This package contains the AI models used by the Docling PDF conversion package
|
5
5
|
License: MIT
|
6
6
|
Keywords: docling,convert,document,pdf,layout model,segmentation,table structure,table former
|
@@ -19,12 +19,13 @@ Classifier: Programming Language :: Python :: 3.11
|
|
19
19
|
Classifier: Programming Language :: Python :: 3.12
|
20
20
|
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
|
21
21
|
Requires-Dist: Pillow (>=10.0.0,<11.0.0)
|
22
|
+
Requires-Dist: huggingface_hub (>=0.23,<1)
|
22
23
|
Requires-Dist: jsonlines (>=3.1.0,<4.0.0)
|
23
24
|
Requires-Dist: lxml (>=4.9.1,<5.0.0)
|
24
25
|
Requires-Dist: mean_average_precision (>=2021.4.26.0,<2022.0.0.0)
|
25
26
|
Requires-Dist: numpy (>=1.24.4,<2.0.0)
|
26
27
|
Requires-Dist: onnxruntime (>=1.16.2,<2.0.0)
|
27
|
-
Requires-Dist: opencv-python-headless (>=4.
|
28
|
+
Requires-Dist: opencv-python-headless (>=4.6.0.66,<5.0.0.0)
|
28
29
|
Requires-Dist: torch (>=2.2.2,<2.3.0) ; sys_platform == "darwin" and platform_machine == "x86_64"
|
29
30
|
Requires-Dist: torch (>=2.2.2,<3.0.0) ; sys_platform != "darwin" or platform_machine != "x86_64"
|
30
31
|
Requires-Dist: torchvision (>=0,<1) ; sys_platform != "darwin" or platform_machine != "x86_64"
|
@@ -110,7 +111,7 @@ Below we list datasets used with their description, source, and ***"TableFormer
|
|
110
111
|
|
111
112
|
## Configuration file
|
112
113
|
|
113
|
-
Example configuration can be
|
114
|
+
Example configuration can be found inside test `tests/test_tf_predictor.py`
|
114
115
|
These are the main sections of the configuration file:
|
115
116
|
|
116
117
|
- `dataset`: The directory for prepared data and the parameters used during the data loading.
|
@@ -128,16 +129,13 @@ You can download the model weights and config files from the links:
|
|
128
129
|
- [TableFormer Checkpoint](https://huggingface.co/ds4sd/docling-models/tree/main/model_artifacts/tableformer)
|
129
130
|
- [beehive_v0.0.5](https://huggingface.co/ds4sd/docling-models/tree/main/model_artifacts/layout/beehive_v0.0.5)
|
130
131
|
|
131
|
-
Place the downloaded files into `tests/test_data/model_artifacts/` directory.
|
132
|
-
|
133
132
|
|
134
133
|
## Inference Tests
|
135
134
|
|
136
|
-
|
135
|
+
You can run the inference tests for the models with:
|
137
136
|
|
138
|
-
First download the model weights (see above), then run:
|
139
137
|
```
|
140
|
-
|
138
|
+
python -m pytest tests/
|
141
139
|
```
|
142
140
|
|
143
141
|
This will also generate prediction and matching visualizations that can be found here:
|
@@ -26,7 +26,7 @@ docling_ibm_models/tableformer/utils/app_profiler.py,sha256=Pb7o1zcikKXh7ninaNt4
|
|
26
26
|
docling_ibm_models/tableformer/utils/mem_monitor.py,sha256=ycZ07fUBVVKKLTVGF54jGPDM2aTkKuZWk1kMbOS0wwQ,6353
|
27
27
|
docling_ibm_models/tableformer/utils/torch_utils.py,sha256=uN0rK9mSXy1ewBnBnILrWebJhhVU4N-XJZBqNiLJwlQ,8893
|
28
28
|
docling_ibm_models/tableformer/utils/utils.py,sha256=8Bxf1rEn977lFbY9NX0r5xh9PvxIRipQZX_EZW92XfA,10980
|
29
|
-
docling_ibm_models-1.
|
30
|
-
docling_ibm_models-1.
|
31
|
-
docling_ibm_models-1.
|
32
|
-
docling_ibm_models-1.
|
29
|
+
docling_ibm_models-1.3.1.dist-info/LICENSE,sha256=mBb7ErEcM8VS9OhiGHnQ2kk75HwPhr54W1Oiz3965MY,1088
|
30
|
+
docling_ibm_models-1.3.1.dist-info/METADATA,sha256=W7euvW9ItpwNBZbYXelzFY_O1sseHNe0HO4sJdO3Hbo,7088
|
31
|
+
docling_ibm_models-1.3.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
|
32
|
+
docling_ibm_models-1.3.1.dist-info/RECORD,,
|
File without changes
|
File without changes
|