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
2026-06-18 09:50:37 -07:00
committed by GitHub
parent c4cd86bb15
commit c3b3336654

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)