yujunjun/olmocr
1
0
Fork 0
mirror of https://github.com/allenai/olmocr.git synced 2025-10-11 16:22:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

testing paddle

Browse Source
This commit is contained in:
Jake Poznanski 2025-09-19 16:55:01 +00:00
parent e9ab2fd1bb
commit 8f75ea062e
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
olmocr/bench/runners/run_paddlepaddle.py
View File

@ -27,6 +27,7 @@ def run_paddlepaddle(
output = pipeline.predict(pdf_path) output = pipeline.predict(pdf_path)
for cur_page_0_indexed, res in enumerate(output): for cur_page_0_indexed, res in enumerate(output):
if cur_page_0_indexed == page_num - 1: if cur_page_0_indexed == page_num - 1:
print(res.markdown)
return res.markdown return res.markdown
raise ValueError(f"Did not get markdown for page {page_num}") raise ValueError(f"Did not get markdown for page {page_num}")