mirror of
https://github.com/DS4SD/docling.git
synced 2025-07-27 04:24:45 +00:00
fix load_from_doctags usage
Signed-off-by: Saidgurbuz <said.gurbuz@epfl.ch>
This commit is contained in:
parent
98b5eeb844
commit
0867936a3a
@ -3,7 +3,7 @@ from io import BytesIO
|
|||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import List, Optional, Union, cast
|
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 import BoundingBox, DocItem, ImageRef, PictureItem, TextItem
|
||||||
from docling_core.types.doc.document import DocTagsDocument
|
from docling_core.types.doc.document import DocTagsDocument
|
||||||
from PIL import Image as PILImage
|
from PIL import Image as PILImage
|
||||||
@ -133,7 +133,7 @@ class VlmPipeline(PaginatedPipeline):
|
|||||||
doctags_doc = DocTagsDocument.from_doctags_and_image_pairs(
|
doctags_doc = DocTagsDocument.from_doctags_and_image_pairs(
|
||||||
doctags_list_c, image_list_c
|
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 forced backend text, replace model predicted text with backend one
|
||||||
if page.size:
|
if page.size:
|
||||||
|
Loading…
Reference in New Issue
Block a user