shenjingwen commited on
Commit
295cea1
ยท
verified ยท
1 Parent(s): 434a82b

Update whisper_asr.py

Browse files
Files changed (1) hide show
  1. whisper_asr.py +2 -1
whisper_asr.py CHANGED
@@ -28,7 +28,8 @@ class WhisperAutomaticSpeechRecognizer:
28
  #hallucination_silence_threshold=0.2 # ๅฏๆ นๆฎๅฎž้™…ๆƒ…ๅ†ต่ฐƒๆ•ด
29
  )
30
  diarize_model = whisperx.DiarizationPipeline(
31
- use_auth_token=os.environ.get('HF_TOKEN'), device="cuda"
 
32
  )
33
  existing_speaker = None
34
 
 
28
  #hallucination_silence_threshold=0.2 # ๅฏๆ นๆฎๅฎž้™…ๆƒ…ๅ†ต่ฐƒๆ•ด
29
  )
30
  diarize_model = whisperx.DiarizationPipeline(
31
+ # use_auth_token=os.environ.get('HF_TOKEN'), device="cuda"
32
+ use_auth_token=os.environ.get('HF_TOKEN'), device=device
33
  )
34
  existing_speaker = None
35