Changeset 36867 for gs3-extensions
- Timestamp:
- 2022-10-28T10:27:21+13:00 (17 months ago)
- Location:
- gs3-extensions/structured-audio/trunk
- Files:
-
- 1 added
- 1 edited
Legend:
- Unmodified
- Added
- Removed
-
gs3-extensions/structured-audio/trunk/bin/script/diarization.py
r36211 r36867 1 #!/usr/bin/env python 1 #!/usr/bin/env python3 2 2 from pyannote.audio import Pipeline 3 3 from pathlib import Path … … 6 6 # remove available graphic cards to trigger cpu default 7 7 os.environ["CUDA_VISIBLE_DEVICES"] = "" 8 9 from huggingface_hub import HfApi 10 available_pipelines = [p.modelId for p in HfApi().list_models(filter="pyannote-audio-pipeline")] 11 print(available_pipelines) 8 12 9 13 parser = argparse.ArgumentParser() … … 27 31 FILEIN = {'audio': fileName} 28 32 fileName, fileExtension = os.path.splitext(fileName) 29 pipeline = Pipeline.from_pretrained( 'pyannote/speaker-diarization')33 pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization") 30 34 diarization = pipeline(FILEIN) 31 35 try:
Note:
See TracChangeset
for help on using the changeset viewer.