Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -338,14 +338,17 @@ class WhisperBase(ABC):
                 total_info += f'Input file:\t\t {info["input_source_file"]}\nLanguage:\t {info["lang"]} (probability {info["lang_prob"]}%)\n'
                 
                 if params.is_translate:
-                    total_info += f'Translation:\t {info["transcription"]}\n'
+                    total_info += f'Translation:\t {info["transcription"]} (①)\n'
                 
                 if translate_output:
-                    total_info += f'Translation:\t {info["translation"]}\n'
+                    total_info += f'Translation:\t {info["translation"]} (②)\n'
             
             time_end = datetime.now()
             total_info += f"\nTotal processing time: {self.format_time((time_end-time_start).total_seconds())}"
 
+            if params.is_translate or translate_output:
+                total_info += "\n\n(Note: ① Translated using OpenAI Whisper, ② Translated using Facebook NLLB)"
+            
             result_str = total_result.rstrip("\n")
             result_file_path = [info['path'] for info in files_to_download.values()]