diff --git a/docling/pipeline/vlm_pipeline.py b/docling/pipeline/vlm_pipeline.py index 9a7b51eb..0e51332c 100644 --- a/docling/pipeline/vlm_pipeline.py +++ b/docling/pipeline/vlm_pipeline.py @@ -3,7 +3,7 @@ from io import BytesIO from pathlib import Path from typing import List, Optional, Union, cast -# from docling_core.types import DoclingDocument +from docling_core.types import DoclingDocument from docling_core.types.doc import BoundingBox, DocItem, ImageRef, PictureItem, TextItem from docling_core.types.doc.document import DocTagsDocument from PIL import Image as PILImage @@ -133,7 +133,7 @@ class VlmPipeline(PaginatedPipeline): doctags_doc = DocTagsDocument.from_doctags_and_image_pairs( doctags_list_c, image_list_c ) - conv_res.document.load_from_doctags(doctags_doc) + conv_res.document = DoclingDocument.load_from_doctags(doctags_doc) # If forced backend text, replace model predicted text with backend one if page.size: