deployment/kaggle/example_inference.py

#!/usr/bin/env python3
"""Run the Kaggle model bundle on one audio file."""

from __future__ import annotations

import argparse
import json
from pathlib import Path

from turn_detector import TurnDetector


def main() -> int:
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument("audio", help="WAV/FLAC/OGG audio file")
    parser.add_argument(
        "--model-dir",
        default=Path(__file__).resolve().parent,
        type=Path,
        help="directory containing model.onnx and model_metadata.json",
    )
    parser.add_argument("--silence-ms", type=float, default=300.0)
    args = parser.parse_args()
    detector = TurnDetector(args.model_dir)
    print(json.dumps(detector.predict_file(args.audio, silence_ms=args.silence_ms), indent=2))
    return 0


if __name__ == "__main__":
    raise SystemExit(main())