Using Docling with costume layout and table recognition models

[ALIYoussef]

Is it possible to use the docling with costume models for layout and table recognition.

I would like to use the pipeline by replacing the existing models with my own models for layout and table recognition. I am wondering if the documentation has any example about using costume AI models.

[dolfim-ibm]

The choice of the models is done at the Pipeline level. For example, the PDF pipeline (called StandardPdfPipeline) is defined in docling/pipeline/standard_pdf_pipeline.py.

You can make your own pipeline with different models, or simply extend with others. We have an example which extends the PDF pipeline with an image understanding model. See https://ds4sd.github.io/docling/examples/develop_picture_enrichment/.

[PeterStaar-IBM]

@ALIYoussef You can of course also provide extension to docling via a PR.