fix(whisperx): use whisperx.diarize.DiarizationPipeline with token kwarg (#10389)

Signed-off-by: Anai-Guo <antai12232931@outlook.com>
This commit is contained in:
Tai An authored and GitHub committed 2026-06-18 18:50:37 +02:00
1 parent c4cd86bb15
commit c3b3336654
1 file changed
+3 -2
+3 -2
View File
@@ -79,6 +79,7 @@ class BackendServicer(backend_pb2_grpc.BackendServicer):
def AudioTranscription(self, request, context):
import whisperx
from whisperx.diarize import DiarizationPipeline
resultSegments = []
text = ""
@@ -106,8 +107,8 @@ class BackendServicer(backend_pb2_grpc.BackendServicer):
# Diarize if requested and HF token is available
if request.diarize and self.hf_token:
if self.diarize_pipeline is None:
self.diarize_pipeline = whisperx.DiarizationPipeline(
use_auth_token=self.hf_token,
self.diarize_pipeline = DiarizationPipeline(
token=self.hf_token,
device=self.device,
)
diarize_segments = self.diarize_pipeline(audio)