Changeset 36867

Timestamp:

2022-10-28T10:27:21+13:00 (18 months ago)

Author:

davidb

Message:

the result of checking out and running this extension using the new self-contained python3

Location:

gs3-extensions/structured-audio/trunk

Files:

• README-NG.txt (added)
• bin/script/diarization.py (modified) (3 diffs)

gs3-extensions/structured-audio/trunk/bin/script/diarization.py

@@ -1 @@
-#!/usr/bin/env python
+#!/usr/bin/env python3
 from pyannote.audio import Pipeline
 from pathlib import Path
@@ -6 +6 @@
 # remove available graphic cards to trigger cpu default
 os.environ["CUDA_VISIBLE_DEVICES"] = "" 
+
+from huggingface_hub import HfApi
+available_pipelines = [p.modelId for p in HfApi().list_models(filter="pyannote-audio-pipeline")]
+print(available_pipelines)
 
 parser = argparse.ArgumentParser()
@@ -27 +31 @@
     FILEIN = {'audio': fileName}
     fileName, fileExtension = os.path.splitext(fileName)
-    pipeline = Pipeline.from_pretrained('pyannote/speaker-diarization')
+    pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization")
     diarization = pipeline(FILEIN)
     try: