Update asr_diarization/pipeline.py

asr_diarization/pipeline.py

@@ -25,7 +25,7 @@ from jiwer import wer, Compose, ToLowerCase, RemovePunctuation, RemoveMultipleSp
 class ASR_Diarization:
     def __init__(self, HF_TOKEN,
                  diar_model="pyannote/speaker-diarization-3.1",
-                 asr_model="openai/whisper-medium"):
+                 asr_model="Capstone04/TrainedWhisper"):
         self.HF_TOKEN = HF_TOKEN
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         self._unknown_lock = threading.Lock()