Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -233,7 +233,8 @@ 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.items(), key=lambda x: x[1]))
+                        max_k, max_v = max(probs.items(), key=lambda x: x[1])
+                        file_lang_probs = str(round(max_v,2))
                         break
 
                 transcribed_segments, time_for_task = self.run(