From e0b82721c7c74392d01460be078068edb10edc4d Mon Sep 17 00:00:00 2001 From: Michele Dolfi Date: Mon, 3 Feb 2025 21:57:45 +0100 Subject: [PATCH] expose picture classifier in CLI Signed-off-by: Michele Dolfi --- docling/cli/main.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/docling/cli/main.py b/docling/cli/main.py index 7d31221d..19f77e4e 100644 --- a/docling/cli/main.py +++ b/docling/cli/main.py @@ -219,6 +219,13 @@ def convert( bool, typer.Option(..., help="Enable the formula enrichment model in the pipeline."), ] = False, + enrich_picture_classes: Annotated[ + bool, + typer.Option( + ..., + help="Enable the picture classification enrichment model in the pipeline.", + ), + ] = False, artifacts_path: Annotated[ Optional[Path], typer.Option(..., help="If provided, the location of the model artifacts."), @@ -375,6 +382,7 @@ def convert( do_table_structure=True, do_code_enrichment=enrich_code, do_formula_enrichment=enrich_formula, + do_picture_classification=enrich_picture_classes, document_timeout=document_timeout, ) pipeline_options.table_structure_options.do_cell_matching = (