mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-12-13 07:08:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Renaming changes

Browse Source 
Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
This commit is contained in:
Christoph Auer
2024-09-27 15:19:35 +02:00
parent 2461b56b84
commit 91ab382129
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docling/datamodel/base_models.py
View File

@@ -5,7 +5,7 @@ from io import BytesIO
from typing import Annotated, Any, Dict, List, Optional, Tuple, Union from typing import Annotated, Any, Dict, List, Optional, Tuple, Union
from docling_core.types.experimental import BoundingBox, Size from docling_core.types.experimental import BoundingBox, Size
from docling_core.types.experimental.document import BaseFigureData, TableCell from docling_core.types.experimental.document import BasePictureData, TableCell
from docling_core.types.experimental.labels import DocItemLabel from docling_core.types.experimental.labels import DocItemLabel
from PIL.Image import Image from PIL.Image import Image
from pydantic import BaseModel, ConfigDict, Field, model_validator from pydantic import BaseModel, ConfigDict, Field, model_validator
@@ -84,7 +84,7 @@ class TextElement(BasePageElement): ...
class FigureElement(BasePageElement): class FigureElement(BasePageElement):
data: Optional[BaseFigureData] = None data: Optional[BasePictureData] = None
provenance: Optional[str] = None provenance: Optional[str] = None
predicted_class: Optional[str] = None predicted_class: Optional[str] = None
confidence: Optional[float] = None confidence: Optional[float] = None

2
docling/models/ds_glm_model.py
View File

@@ -57,7 +57,7 @@ class GlmModel:
if arr == "tables": if arr == "tables":
prov = ds_document.tables[index].prov[0] prov = ds_document.tables[index].prov[0]
elif arr == "figures": elif arr == "figures":
prov = ds_document.figures[index].prov[0] prov = ds_document.pictures[index].prov[0]
else: else:
prov = None prov = None