mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-08-02 15:32:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

expose picture classifier in CLI

Browse Source 
Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>
This commit is contained in:
Michele Dolfi 2025-02-03 21:57:45 +01:00
parent b5da4080c9
commit e0b82721c7
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
docling/cli/main.py
View File

@ -219,6 +219,13 @@ def convert(
bool, bool,
typer.Option(..., help="Enable the formula enrichment model in the pipeline."), typer.Option(..., help="Enable the formula enrichment model in the pipeline."),
] = False, ] = False,
enrich_picture_classes: Annotated[
bool,
typer.Option(
...,
help="Enable the picture classification enrichment model in the pipeline.",
),
] = False,
artifacts_path: Annotated[ artifacts_path: Annotated[
Optional[Path], Optional[Path],
typer.Option(..., help="If provided, the location of the model artifacts."), typer.Option(..., help="If provided, the location of the model artifacts."),
@ -375,6 +382,7 @@ def convert(
do_table_structure=True, do_table_structure=True,
do_code_enrichment=enrich_code, do_code_enrichment=enrich_code,
do_formula_enrichment=enrich_formula, do_formula_enrichment=enrich_formula,
do_picture_classification=enrich_picture_classes,
document_timeout=document_timeout, document_timeout=document_timeout,
) )
pipeline_options.table_structure_options.do_cell_matching = ( pipeline_options.table_structure_options.do_cell_matching = (