clean up logic, use pandas where possibl

i4Ds · Jan 25, 2023 · 286a2f2 · 286a2f2
1 parent eec6d1f
commit 286a2f2
Show file tree

Hide file tree

Showing 5 changed files with 425 additions and 428 deletions.
diff --git a/EXAMPLES.md b/EXAMPLES.md
@@ -2,7 +2,7 @@
 
 ## Other Languages
 
-For non-english ASR, it is best to use the `large` whisper model. Alignment models are automatically picked by the chosen language from the default [lists](https://github.com/m-bain/whisperX/blob/e909f2f766b23b2000f2d95df41f9b844ac53e49/whisperx/transcribe.py#L22).
+For non-english ASR, it is best to use the `large` whisper model. Alignment models are automatically picked by the chosen language from the default [lists](https://github.com/m-bain/whisperX/blob/main/whisperx/alignment.py#L18).
 
 Currently support default models tested for {en, fr, de, es, it, ja, zh, nl}
 

diff --git a/whisperx/__init__.py b/whisperx/__init__.py
@@ -11,8 +11,8 @@
 from .audio import load_audio, log_mel_spectrogram, pad_or_trim
 from .decoding import DecodingOptions, DecodingResult, decode, detect_language
 from .model import Whisper, ModelDimensions
-from .transcribe import transcribe, load_align_model, align, transcribe_with_vad
-
+from .transcribe import transcribe, transcribe_with_vad
+from .alignment import load_align_model, align
 _MODELS = {
     "tiny.en": "https://openaipublic.azureedge.net/main/whisper/models/d3dd57d32accea0b295c96e26691aa14d8822fac7d9d27d5dc00b4ca2826dd03/tiny.en.pt",
     "tiny": "https://openaipublic.azureedge.net/main/whisper/models/65147644a518d12f04e32d6f3b26facc3f8dd46e5390956a9424a650c0ce22b9/tiny.pt",