From 99c1df9bc18a8d0c4adf4515a65418eff9ba6e2e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Drago=C8=99?= <balandragos5555@gmail.com>
Date: Thu, 18 Apr 2024 15:35:35 +0200
Subject: [PATCH] Add faster-whisper version used to provenance

---
 whisper.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/whisper.py b/whisper.py
index 57ea51a..8a18c64 100644
--- a/whisper.py
+++ b/whisper.py
@@ -13,7 +13,7 @@
     get_base_output_dir,
     get_output_file_path,
 )
-from faster_whisper import WhisperModel
+import faster_whisper
 
 
 def run_whisper(
@@ -25,7 +25,7 @@ def run_whisper(
     destination = get_output_file_path(input.source_id, OutputType.TRANSCRIPT)
 
     # float16 only works on GPU, float32 or int8 are recommended for CPU
-    model = WhisperModel(
+    model = faster_whisper.WhisperModel(
         cfg.WHISPER_ASR_SETTINGS.MODEL_VERSION,
         device=cfg.WHISPER_ASR_SETTINGS.DEVICE,
         compute_type=(
@@ -85,6 +85,7 @@ def run_whisper(
         input_data=input.input_file_path,
         start_time_unix=start,
         parameters=cfg.WHISPER_ASR_SETTINGS,
+        software_version=faster_whisper.__version__,
         output_data=destination,
         processing_time_ms=end - start,
     )