fixing whisper

TalkBank · Feb 23, 2024 · 7143ecc · 7143ecc
1 parent 0a5b848
commit 7143ecc
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 9 deletions.
diff --git a/batchalign/pipelines/asr/whisperx.py b/batchalign/pipelines/asr/whisperx.py
@@ -30,7 +30,7 @@ def __init__(self, lang="eng"):
 
         try:
             import whisperx
-
+            print("Batchalign: If you got a warning about pyannote versioning, bad things will not happen. Please disregard.")
         except ImportError:
             raise ImportError("Cannot import WhisperX, please ensure it is installed.\nHint: install WhisperX by running `pip install git+https://github.com/m-bain/whisperx.git`.")
 
@@ -83,12 +83,13 @@ def generate(self, source_path, **kwargs):
             for word in segment["words"]:
                 stripped = word["word"].translate(str.maketrans('', '', string.punctuation)).strip()
                 if stripped != "":
-                    text = {
-                        "type": "text",
-                        "ts": word["start"],
-                        "end_ts": word["end"],
-                        "value": stripped,
-                    }
+                    if word.get("start") != None and word.get("end") != None:
+                        text = {
+                            "type": "text",
+                            "ts": word["start"],
+                            "end_ts": word["end"],
+                            "value": stripped,
+                        }
 
                     current_turn.append(text)
 

diff --git a/batchalign/version b/batchalign/version
@@ -1,3 +1,3 @@
-0.5.2-beta.0
+0.5.2-beta.1
 Feb 16th, 2024
-Disable MWT for GA
+Whisperx fixes