Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -221,6 +221,8 @@ class WhisperBase(ABC):
             
             files_info = {}
             files_to_download = {}
+            time_start = datetime.now()
+            
             for file in files:
 
                 ## Detect language
@@ -284,7 +286,10 @@ class WhisperBase(ABC):
                 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"\nTranscription duration: {self.format_time(total_time)}"
+            #total_info += f"\nTranscription duration: {self.format_time(total_time)}"
+            time_end = datetime.now()
+            total_info += f"\nTranscription duration: {self.format_time(time_end-time_start)}"
+            
             result_str = total_result
             result_file_path = [info['path'] for info in files_to_download.values()]