23 lines
615 B
Python
23 lines
615 B
Python
"""Internal entry point for the isolated pyannote container adapter."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
from pathlib import Path
|
|
|
|
from src.meeting_lab.diarization.backend import run_native_pyannote
|
|
|
|
|
|
def main() -> int:
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("audio", type=Path)
|
|
parser.add_argument("output", type=Path)
|
|
parser.add_argument("--device", choices=("auto", "gpu", "cpu"), required=True)
|
|
args = parser.parse_args()
|
|
run_native_pyannote(args.audio, args.output, args.device)
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|