Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -249,7 +249,7 @@ class WhisperBase(ABC):
                     file_format=file_format,
                     output_dir=self.output_dir
                 )
-                files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language, "input_source_file": (file_name+file_ext)}
+                files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language, "lang_prob": str(probs), "input_source_file": (file_name+file_ext)}
 
                 ## Add output file as txt
                 file_name, file_ext = os.path.splitext(os.path.basename(file))
@@ -279,7 +279,7 @@ class WhisperBase(ABC):
             for file_name, info in files_info.items():
                 total_result += f'{info["subtitle"]}'
                 total_time += info["time_for_task"]
-                total_info += f'Input file: {info["input_source_file"]}\nLanguage prediction: {info["lang"]}\n'
+                total_info += f'Input file: {info["input_source_file"]}\nLanguage prediction: {info["lang"]} with probability {info["lang_prob"]}\n'
             
             total_info += f"\nTranscription duration: {self.format_time(total_time)}"
             result_str = total_result