Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -236,7 +236,7 @@ class WhisperBase(ABC):
                     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,2))*100)
+                        file_lang_probs = str((round(max_v*100)))
                         break
     
                 transcribed_segments, time_for_task = self.run(
@@ -284,7 +284,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"]} with probability {info["lang_prob"]}%\n'
+                total_info += f'Input file: {info["input_source_file"]}\nLanguage: {info["lang"]} with probability {info["lang_prob"]}%\n'
             
             #total_info += f"\nTranscription duration: {self.format_time(total_time)}"
             time_end = datetime.now()