Update modules/whisper/whisper_base.py
Browse files
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 |
-
|
237 |
-
|
238 |
-
|
239 |
-
|
240 |
-
|
241 |
-
|
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(
|