Update asr_diarization/pipeline.py
Browse files
asr_diarization/pipeline.py
CHANGED
|
@@ -14,7 +14,7 @@ from jiwer import wer, Compose, ToLowerCase, RemovePunctuation, RemoveMultipleSp
|
|
| 14 |
class ASR_Diarization:
|
| 15 |
def __init__(self, HF_TOKEN,
|
| 16 |
diar_model="pyannote/speaker-diarization-3.1",
|
| 17 |
-
asr_model="Capstone04/
|
| 18 |
model_path=None): # NEW: model_path parameter
|
| 19 |
self.HF_TOKEN = HF_TOKEN
|
| 20 |
self.device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
|
| 14 |
class ASR_Diarization:
|
| 15 |
def __init__(self, HF_TOKEN,
|
| 16 |
diar_model="pyannote/speaker-diarization-3.1",
|
| 17 |
+
asr_model="Capstone04/TrainedWhisper_Medium",
|
| 18 |
model_path=None): # NEW: model_path parameter
|
| 19 |
self.HF_TOKEN = HF_TOKEN
|
| 20 |
self.device = "cuda" if torch.cuda.is_available() else "cpu"
|