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(
|