Display processing time and progress indicator

main.py

 from faster_whisper import WhisperModel
 from datetime import timedelta
+from time import perf_counter
 
 model_size = "large-v2"
 
+tStart = perf_counter()
+
 # Run on GPU with FP32
 model = WhisperModel(model_size, device="cuda", compute_type="float32")
+print("Model initialized")
 
 # Run on CPU
 # model = WhisperModel(model_size, device="cpu", compute_type="float32")
@@ -14,6 +18,7 @@ segments, info = model.transcribe("/input/audiofile", language="de", beam_size=5
 with open("/output/transcript.vtt", "w", encoding="utf-8") as f:
     f.write("WEBVTT\n\nNOTE This transcript was automatically generated.")
 
+    print("Processing...", end="")
     for segment in segments:
         start = timedelta(seconds=segment.start)
         end = timedelta(seconds=segment.end)
@@ -21,7 +26,15 @@ with open("/output/transcript.vtt", "w", encoding="utf-8") as f:
 
         entry = "{} --> {}\n{}".format(start, end, text)
 
-        print(entry)
-
         f.write("\n\n")
         f.write(entry)
+
+        progress = segment.end / info.duration
+        print("{:2.0%}".format(progress if progress <= 1 else 1), end="...")
+
+print("Done!")
+
+tEnd = perf_counter()
+tDelta = tEnd - tStart
+
+print("Processing time: {}s".format(timedelta(seconds=tDelta)))