Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -347,7 +347,13 @@ class WhisperBase(ABC):
             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\nNote: translation handled by OpenAI Whisper ¹ or Facebook NLLB ²"
+                #total_info += "\n\nNote: translation handled by OpenAI Whisper ¹ or Facebook NLLB ²"
+                if params.is_translate and not translate_output:
+                    total_info += "\n\nNote: translation handled by OpenAI Whisper"
+                if not params.is_translate and translate_output:
+                    total_info += "\n\nNote: translation handled by Facebook NLLB"
+                if params.is_translate and translate_output:    
+                    total_info += "\n\nNote: translation handled by OpenAI Whisper or Facebook NLLB"
             
             result_str = total_result.rstrip("\n")
             result_file_path = [info['path'] for info in files_to_download.values()]