mirror of
https://github.com/allenai/olmocr.git
synced 2025-10-10 07:42:13 +00:00
testing paddle
This commit is contained in:
parent
e9ab2fd1bb
commit
8f75ea062e
@ -27,6 +27,7 @@ def run_paddlepaddle(
|
||||
output = pipeline.predict(pdf_path)
|
||||
for cur_page_0_indexed, res in enumerate(output):
|
||||
if cur_page_0_indexed == page_num - 1:
|
||||
print(res.markdown)
|
||||
return res.markdown
|
||||
|
||||
raise ValueError(f"Did not get markdown for page {page_num}")
|
||||
|
Loading…
x
Reference in New Issue
Block a user