mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-07-30 14:04:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix for the tables when using OCR

Browse Source 
Signed-off-by: Maksym Lysak <mly@zurich.ibm.com>
This commit is contained in:
Maksym Lysak 2025-03-29 09:33:54 +01:00
parent 124f921077
commit d3c10b24a4
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
docling/models/table_structure_model.py
View File

@ -229,6 +229,9 @@ class TableStructureModel(BasePageModel):
cell_unit=TextCellUnit.WORD, cell_unit=TextCellUnit.WORD,
bbox=table_cluster.bbox, bbox=table_cluster.bbox,
) )
if len(tcells) == 0:
# In case word-level cells yield empty
tcells = table_cluster.cells
else: else:
# Otherwise - we use normal (line/phrase) cells # Otherwise - we use normal (line/phrase) cells
tcells = table_cluster.cells tcells = table_cluster.cells