bumping increasing

batchalign/pipelines/asr/whisperx.py

@@ -16,12 +16,6 @@
 import io
 import os
 
-try:
-    import whisperx
-
-except ImportError:
-    raise ImportError("Cannot import WhisperX, please ensure it is installed.\nHint: run `pip install -U batchalign[whisperx]` or install WhisperX via instructions (https://github.com/m-bain/whisperX).")
-
 import gc 
 import torch
 
@@ -34,6 +28,14 @@ class WhisperXEngine(BatchalignEngine):
 
     def __init__(self, lang="eng"):
 
+        try:
+            import whisperx
+
+        except ImportError:
+            raise ImportError("Cannot import WhisperX, please ensure it is installed.\nHint: install WhisperX via instructions (https://github.com/m-bain/whisperX).")
+
+
+
         language = pycountry.languages.get(alpha_3=lang).alpha_2
 
         self.__lang = lang
@@ -56,6 +58,14 @@ def __init__(self, lang="eng"):
             self.__engine = None
 
     def generate(self, source_path, **kwargs):
+
+        try:
+            import whisperx
+
+        except ImportError:
+            raise ImportError("Cannot import WhisperX, please ensure it is installed.\nHint: run `pip install -U batchalign[whisperx]` or install WhisperX via instructions (https://github.com/m-bain/whisperX).")
+
+
         # load audio
         audio = whisperx.load_audio(source_path)
 

setup.py

@@ -55,9 +55,6 @@ def read(fname):
     extras_require={
         'dev': [
             'pytest',
-        ],
-        "whisperx": [
-            "whisperx @ git+ssh://[email protected]:m-bain/whisperX.git"
         ]
     },
     include_package_data=True,