Update tools/extract/asr.py

tools/extract/asr.py

@@ -21,7 +21,15 @@ class ASRProcessor:
 
     def __init__(self, model_name="large-v2", compute_type="float16"):
         self.model_name = model_name
-        self.model = whisperx.load_model(model_name, device, compute_type=compute_type)
+        # Check if whisperx.load_model accepts compute_type parameter
+        import inspect
+
+        if "compute_type" in inspect.signature(whisperx.load_model).parameters:
+            self.model = whisperx.load_model(
+                model_name, device, compute_type=compute_type
+            )
+        else:
+            self.model = whisperx.load_model(model_name, device)
 
     def get_asr(self, audio_file, return_duration=True):
         assert Path(audio_file).exists(), f"File {audio_file} does not exist"