| """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() | |