Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -338,14 +338,19 @@ 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 += "Note:"
+                total_info += "\t①: using OpenAI Whisper\n"
+                total_info += "\t②: using Facebook NLLB\n"
+
             result_str = total_result.rstrip("\n")
             result_file_path = [info['path'] for info in files_to_download.values()]