diff --git a/docling/models/layout_model.py b/docling/models/layout_model.py index af7b8e7b..c5cb0bef 100644 --- a/docling/models/layout_model.py +++ b/docling/models/layout_model.py @@ -33,6 +33,7 @@ class LayoutModel: "Page-footer", "Code", "List-item", + # "Title" # "Formula", ] PAGE_HEADER_LABELS = ["Page-header", "Page-footer"] @@ -71,6 +72,7 @@ class LayoutModel: CLASS_REMAPPINGS = { "Document Index": "Table", + "Title": "Section-header" } _log.debug("================= Start postprocess function ====================") @@ -277,6 +279,7 @@ class LayoutModel: bbox=BoundingBox.model_validate(pred_item), cells=[], ) + clusters.append(cluster) # Map cells to clusters