Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -259,16 +259,6 @@ class WhisperBase(ABC):
                     *whisper_params,
                 )
                 
-                #file_name, file_ext = os.path.splitext(os.path.basename(file))
-                #subtitle, file_path = self.generate_and_write_file(
-                #    file_name=file_name,
-                #    transcribed_segments=transcribed_segments,
-                #    add_timestamp=add_timestamp,
-                #    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, "lang_prob": file_lang_probs, "input_source_file": (file_name+file_ext)}
-
                 # Define source language
                 source_lang = file_language
                 
@@ -293,7 +283,7 @@ class WhisperBase(ABC):
                         if source_lang != target_lang:
                             translation_note = source_lang + " to " + target_lang
                         else:
-                            translation_note = source_lang + " equal to " + target_lang
+                            translation_note = "Source & target languages are " + target_lang
                     else:
                         translation_note = source_lang + " not supported"
                 
@@ -341,7 +331,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: {info["lang"]} with probability {info["lang_prob"]}%\n'
+                total_info += f'Input file: {info["input_source_file"]}\nLanguage: {info["lang"]} (probability {info["lang_prob"]}%)\n'
                 
                 if params.is_translate:
                     total_info += f'Transcription: {info["transcription"]}\n'