Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -233,11 +233,19 @@ class WhisperBase(ABC):
                 file_language = ""
                 file_lang_probs = ""
                 for key,value in whisper.tokenizer.LANGUAGES.items():
-                    if key == str(max(probs, key=probs.get)):
-                        file_language = value.capitalize()
-                        max_k, max_v = max(probs.items(), key=lambda x: x[1])
-                        file_lang_probs = str((round(max_v*100)))
-                        break
+                	if key == str(max(probs, key=probs.get)):
+                		file_language = value.capitalize()
+                		for key_prob,value_prob in probs.items():
+                			if key == key_prob:
+                				file_lang_probs = str((round(value_prob*100)))
+                				break
+                	break
+#                for key,value in whisper.tokenizer.LANGUAGES.items():
+#                    if key == str(max(probs, key=probs.get)):
+#                        file_language = value.capitalize()
+#                        max_k, max_v = max(probs.items(), key=lambda x: x[1])
+#                        file_lang_probs = str((round(max_v*100)))
+#                        break
     
                 transcribed_segments, time_for_task = self.run(
                     file,