mirror of
https://github.com/DS4SD/docling.git
synced 2025-07-26 20:14:47 +00:00
Renaming changes
Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
This commit is contained in:
parent
2461b56b84
commit
91ab382129
@ -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
|
||||
|
@ -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
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user