Forensic-Noise-Classifier / utils /vad_segmentation.py
Knight-coderr's picture
Update utils/vad_segmentation.py
4390902 verified
raw
history blame contribute delete
533 Bytes
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)