LAP-DEV commited on
Commit
20cd2c1
·
verified ·
1 Parent(s): 15bef2f

Update modules/whisper/whisper_base.py

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +1 -1
modules/whisper/whisper_base.py CHANGED
@@ -233,7 +233,7 @@ class WhisperBase(ABC):
233
  for key,value in whisper.tokenizer.LANGUAGES.items():
234
  if key == str(max(probs, key=probs.get)):
235
  file_language = value.capitalize()
236
- file_lang_probs = probs #str(max(probs, key=lambda key: probs[1]))
237
  break
238
 
239
  transcribed_segments, time_for_task = self.run(
 
233
  for key,value in whisper.tokenizer.LANGUAGES.items():
234
  if key == str(max(probs, key=probs.get)):
235
  file_language = value.capitalize()
236
+ file_lang_probs = str(max(probs.items(), key=lambda x: x[1]))
237
  break
238
 
239
  transcribed_segments, time_for_task = self.run(