INNER CODE UNIT · Python

main

mindee/doctr · doctr/cli/main.py:315

def main(argv=None):
    """Main function for the docTR CLI tool"""
    # parse command-line arguments and set up the model
    args = _parse_args(argv)
    _setup_logging(args.quiet)

    fmt = _resolve_format(args)
    model = _build_predictor(args)

    # load the document
    doc = _load_document(args)

    # perform OCR
    logger.info("Performing OCR...")
    result = model(doc)

    # save results to the requested format
    _save_results(result, fmt, args)

View source record →