fix load_from_doctags usage

Signed-off-by: Saidgurbuz <said.gurbuz@epfl.ch>
This commit is contained in:
Saidgurbuz 2025-05-20 09:42:34 +02:00
parent 98b5eeb844
commit 0867936a3a

View File

@ -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: