Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -233,7 +233,7 @@ class WhisperBase(ABC):
                 for key,value in whisper.tokenizer.LANGUAGES.items():
                     if key == str(max(probs, key=probs.get)):
                         file_language = value.capitalize()
-                        file_lang_probs = str(max(probs, key=lambda key: probs[1]))
+                        file_lang_probs = probs #str(max(probs, key=lambda key: probs[1]))
                         break
 
                 transcribed_segments, time_for_task = self.run(