mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-12-08 12:48:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: Add more logs in LayoutModel

Browse Source 
Signed-off-by: Nikos Livathinos <nli@zurich.ibm.com>
This commit is contained in:
Nikos Livathinos
2025-08-28 14:24:47 +02:00
parent baaf2698b4
commit 36d44f1225
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
docling/models/layout_model.py
View File

@@ -188,6 +188,11 @@ class LayoutModel(BasePageModel):
clusters = []
for ix, pred_item in enumerate(page_predictions):
#################################################################################
# Debug: Print the un-processed clusters
#
print(f"{ix}: Pred item: {pred_item}")
#################################################################################
label = DocItemLabel(
pred_item["label"].lower().replace(" ", "_").replace("-", "_")
) # Temporary, until docling-ibm-model uses docling-core types