yujunjun/olmocr
1
0
Fork 0
mirror of https://github.com/allenai/olmocr.git synced 2025-09-09 08:40:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix for anchor generation on pdfs with no text elements

Browse Source
This commit is contained in:
Jake Poznanski 2024-10-11 15:01:01 +00:00
parent af03358c47
commit aea3f7f1fe
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pdelfin/prompts/anchor.py
View File

@ -302,6 +302,7 @@ def _linearize_pdf_report(report: PageReport, max_length: int = 4000) -> str:
if report.text_elements:
text_elements = [e for e in report.text_elements if len(e.text.strip()) > 0]
if text_elements:
min_x_text = min(text_elements, key=lambda e: e.x)
max_x_text = max(text_elements, key=lambda e: e.x)
min_y_text = min(text_elements, key=lambda e: e.y)