fix: Adjust the LayoutModel default paths for the docling-layout-heron

Signed-off-by: Nikos Livathinos <nli@zurich.ibm.com>
This commit is contained in:
Nikos Livathinos 2025-04-15 23:29:01 +02:00
parent e5f8bb086d
commit 4ce338f455

View File

@ -23,8 +23,8 @@ _log = logging.getLogger(__name__)
class LayoutModel(BasePageModel): class LayoutModel(BasePageModel):
_model_repo_folder = "ds4sd--docling-models" _model_repo_folder = "ds4sd--docling-layout-heron"
_model_path = "model_artifacts/layout" _model_path = ""
TEXT_ELEM_LABELS = [ TEXT_ELEM_LABELS = [
DocItemLabel.TEXT, DocItemLabel.TEXT,