Skip to content

Commit

Permalink
feat: expose code and formula models in the CLI
Browse files Browse the repository at this point in the history
Signed-off-by: Michele Dolfi <[email protected]>
  • Loading branch information
dolfim-ibm committed Jan 27, 2025
1 parent a112d7a commit 9c6d922
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions docling/cli/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -211,6 +211,14 @@ def convert(
TableFormerMode,
typer.Option(..., help="The mode to use in the table structure model."),
] = TableFormerMode.FAST,
enrich_code: Annotated[
bool,
typer.Option(..., help="Enable the code enrichment model in the pipeline."),
] = False,
enrich_formula: Annotated[
bool,
typer.Option(..., help="Enable the formula enrichment model in the pipeline."),
] = False,
artifacts_path: Annotated[
Optional[Path],
typer.Option(..., help="If provided, the location of the model artifacts."),
Expand Down Expand Up @@ -365,6 +373,8 @@ def convert(
do_ocr=ocr,
ocr_options=ocr_options,
do_table_structure=True,
do_code_enrichment=enrich_code,
do_formula_enrichment=enrich_formula,
document_timeout=document_timeout,
)
pipeline_options.table_structure_options.do_cell_matching = (
Expand Down

0 comments on commit 9c6d922

Please sign in to comment.