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

Updated minimal vlm pipeline example

Browse Source 
Signed-off-by: Maksym Lysak <mly@zurich.ibm.com>
This commit is contained in:
Maksym Lysak 2025-03-19 13:48:44 +01:00
parent 0875388ce6
commit 9182d8a622
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
docs/examples/minimal_vlm_pipeline.py
View File

@ -17,7 +17,8 @@ from docling.document_converter import DocumentConverter, PdfFormatOption
from docling.pipeline.vlm_pipeline import VlmPipeline from docling.pipeline.vlm_pipeline import VlmPipeline
sources = [ sources = [
"tests/data/2305.03393v1-pg9-img.png", # "tests/data/2305.03393v1-pg9-img.png",
"tests/data/pdf/2305.03393v1-pg9.pdf",
] ]
## Use experimental VlmPipeline ## Use experimental VlmPipeline
@ -73,10 +74,12 @@ for source in sources:
print("") print("")
print(res.document.export_to_markdown()) print(res.document.export_to_markdown())
doctags = ""
for page in res.pages: for page in res.pages:
print("") print("")
print("Predicted page in DOCTAGS:") print("Predicted page in DOCTAGS:")
print(page.predictions.vlm_response.text) print(page.predictions.vlm_response.text)
doctags += page.predictions.vlm_response.text
res.document.save_as_html( res.document.save_as_html(
filename=Path("{}/{}.html".format(out_path, res.input.file.stem)), filename=Path("{}/{}.html".format(out_path, res.input.file.stem)),
@ -87,6 +90,12 @@ for source in sources:
with (out_path / f"{res.input.file.stem}.json").open("w") as fp: with (out_path / f"{res.input.file.stem}.json").open("w") as fp:
fp.write(json.dumps(res.document.export_to_dict())) fp.write(json.dumps(res.document.export_to_dict()))
with (out_path / f"{res.input.file.stem}.md").open("w") as fp:
fp.write(res.document.export_to_markdown())
with (out_path / f"{res.input.file.stem}.doctag").open("w") as fp:
fp.write(doctags)
pg_num = res.document.num_pages() pg_num = res.document.num_pages()
print("") print("")