| """Command-line scoring: python predict.py --audio a.wav [b.ogg ...] [--model-dir .] [--device cpu] | |
| Prints one JSON line per file. score = bona fide likelihood in [0, 1] (uncalibrated).""" | |
| import argparse | |
| import json | |
| from vouch_detector import VouchDetector | |
| def main(): | |
| ap = argparse.ArgumentParser() | |
| ap.add_argument("--audio", nargs="+", required=True) | |
| ap.add_argument("--model-dir", default=".") | |
| ap.add_argument("--device", default="cpu") | |
| args = ap.parse_args() | |
| det = VouchDetector(args.model_dir, device=args.device) | |
| for a in args.audio: | |
| print(json.dumps({"audio": a, **det.score_file(a)})) | |
| if __name__ == "__main__": | |
| main() | |