asr-inference

Running on Zero

ssolito commited on Dec 4, 2024

Commit

551bdb6

verified ·

1 Parent(s): ecb7cc4

Update whisper.py

Files changed (1) hide show

whisper.py CHANGED Viewed

@@ -17,7 +17,7 @@ CKPT = "projecte-aina/whisper-large-v3-tiny-caesar"
 BATCH_SIZE = 1
 model = WhisperForConditionalGeneration.from_pretrained(MODEL_NAME, torch_dtype=torch_dtype).to(device)
 processor = WhisperProcessor.from_pretrained(MODEL_NAME)
-pipeline_vad = Pipeline.from_pretrained("pyannote/voice-activity-detection", use_auth_token=os.getenv("HF_TOKEN"))
 threshold = 10000
 segments_dir = "."

 BATCH_SIZE = 1
 model = WhisperForConditionalGeneration.from_pretrained(MODEL_NAME, torch_dtype=torch_dtype).to(device)
 processor = WhisperProcessor.from_pretrained(MODEL_NAME)
+pipeline_vad = Pipeline.from_pretrained("./pyannote/config.yaml")
 threshold = 10000
 segments_dir = "."