scripts/inference.py

"""Run inference with the locally trained model."""

import argparse
from local_pipeline import infer, load_config


def main() -> None:
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument(
        "--config", default="conf/config.yaml", help="Pipeline YAML configuration",
    )
    args = parser.parse_args()
    infer(load_config(args.config))


if __name__ == "__main__":
    main()