Upload folder using huggingface_hub
Browse files
asr_diarization/pipeline.py
CHANGED
|
@@ -13,7 +13,7 @@ from jiwer import wer, Compose, ToLowerCase, RemovePunctuation, RemoveMultipleSp
|
|
| 13 |
|
| 14 |
|
| 15 |
class ASR_Diarization:
|
| 16 |
-
def __init__(self, HF_TOKEN,
|
| 17 |
diar_model="pyannote/speaker-diarization-3.1",
|
| 18 |
asr_model="Capstone04/TrainedWhisper"):
|
| 19 |
self.HF_TOKEN = HF_TOKEN
|
|
|
|
| 13 |
|
| 14 |
|
| 15 |
class ASR_Diarization:
|
| 16 |
+
def __init__(self, HF_TOKEN, HF_TOKEN1,
|
| 17 |
diar_model="pyannote/speaker-diarization-3.1",
|
| 18 |
asr_model="Capstone04/TrainedWhisper"):
|
| 19 |
self.HF_TOKEN = HF_TOKEN
|