LAP-DEV commited on
Commit
81ca3d5
·
verified ·
1 Parent(s): 1975678

Update modules/whisper/whisper_base.py

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +6 -6
modules/whisper/whisper_base.py CHANGED
@@ -233,12 +233,12 @@ class WhisperBase(ABC):
233
  file_language = ""
234
  file_lang_probs = ""
235
  for key,value in whisper.tokenizer.LANGUAGES.items():
236
- if key == str(max(probs, key=probs.get)):
237
- file_language = value.capitalize()
238
- for key_prob,value_prob in probs.items():
239
- if key == key_prob:
240
- file_lang_probs = str((round(value_prob*100)))
241
- break
242
  break
243
 
244
  transcribed_segments, time_for_task = self.run(
 
233
  file_language = ""
234
  file_lang_probs = ""
235
  for key,value in whisper.tokenizer.LANGUAGES.items():
236
+ if key == str(max(probs, key=probs.get)):
237
+ file_language = value.capitalize()
238
+ for key_prob,value_prob in probs.items():
239
+ if key == key_prob:
240
+ file_lang_probs = str((round(value_prob*100)))
241
+ break
242
  break
243
 
244
  transcribed_segments, time_for_task = self.run(