Merge branch 'main' of https://huggingface.co/spaces/aadnk/whisper-webui into main

src/vad.py

@@ -205,11 +205,16 @@ class AbstractTranscription(ABC):
 
                 print("Running whisper from ", format_timestamp(segment_start), " to ", format_timestamp(segment_end), ", duration: ", 
                     segment_duration, "expanded: ", segment_expand_amount, "prompt: ", segment_prompt, "language: ", detected_language)
-                
+
+                perf_start_time = time.perf_counter()
+
                 scaled_progress_listener = SubTaskProgressListener(progressListener, base_task_total=progress_total_duration, 
                                                                    sub_task_start=segment_start - progress_start_offset, sub_task_total=segment_duration) 
                 segment_result = whisperCallable.invoke(segment_audio, segment_index, segment_prompt, detected_language, progress_listener=scaled_progress_listener)
 
+                perf_end_time = time.perf_counter()
+                print("Whisper took {} seconds".format(perf_end_time - perf_start_time))
+
                 adjusted_segments = self.adjust_timestamp(segment_result["segments"], adjust_seconds=segment_start, max_source_time=segment_duration)
 
                 # Propagate expand amount to the segments

src/whisper/fasterWhisperContainer.py

@@ -7,6 +7,7 @@ from src.hooks.progressListener import ProgressListener
 from src.languages import get_language_from_name
 from src.modelCache import ModelCache
 from src.whisper.abstractWhisperContainer import AbstractWhisperCallback, AbstractWhisperContainer
+from src.utils import format_timestamp
 
 class FasterWhisperContainer(AbstractWhisperContainer):
     def __init__(self, model_name: str, device: str = None, compute_type: str = "float16",
@@ -138,7 +139,8 @@ class FasterWhisperCallback(AbstractWhisperCallback):
             if progress_listener is not None:
                 progress_listener.on_progress(segment.end, info.duration)
             if verbose:
-                print(segment.text)
+                print("[{}->{}] {}".format(format_timestamp(segment.start, True), format_timestamp(segment.end, True),
+                                          segment.text))
 
         text = " ".join([segment.text for segment in segments])