Update utils/vad_segmentation.py

utils/vad_segmentation.py

@@ -1,19 +1,19 @@
-import os
-from pyannote.audio import Model
-from pyannote.audio.pipelines import VoiceActivityDetection
-hf_token = os.environ.get("HF_TOKEN")
-
-model = Model.from_pretrained("pyannote/segmentation", 
-                              use_auth_token=hf_token)
-
-
-vad_pipeline = VoiceActivityDetection(segmentation=model)
-HYPER_PARAMETERS = {
-  "onset": 0.5, "offset": 0.5,
-  "min_duration_on": 0.0,
-  "min_duration_off": 0.0
-}
-
-vad_pipeline.instantiate(HYPER_PARAMETERS)
-def vad_segmentation(input_path, output_path, aggressiveness=2):
-    return vad_pipeline(input_path)
+import os
+from pyannote.audio import Model
+from pyannote.audio.pipelines import VoiceActivityDetection
+hf_token = os.environ.get("HF_TOKEN")
+
+model = Model.from_pretrained("pyannote/segmentation", 
+                              use_auth_token=hf_token)
+
+
+vad_pipeline = VoiceActivityDetection(segmentation=model)
+HYPER_PARAMETERS = {
+  "onset": 0.5, "offset": 0.5,
+  "min_duration_on": 0.0,
+  "min_duration_off": 0.0
+}
+
+vad_pipeline.instantiate(HYPER_PARAMETERS)
+def vad_segmentation(input_path):
+    return vad_pipeline(input_path)