Renaming changes

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

View File

@ -5,7 +5,7 @@ from io import BytesIO
from typing import Annotated, Any, Dict, List, Optional, Tuple, Union
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 PIL.Image import Image
from pydantic import BaseModel, ConfigDict, Field, model_validator
@ -84,7 +84,7 @@ class TextElement(BasePageElement): ...
class FigureElement(BasePageElement):
data: Optional[BaseFigureData] = None
data: Optional[BasePictureData] = None
provenance: Optional[str] = None
predicted_class: Optional[str] = None
confidence: Optional[float] = None

View File

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