mirror of
https://github.com/m-bain/whisperX.git
synced 2025-07-01 18:17:27 -04:00
32 lines
734 B
Python
32 lines
734 B
Python
import importlib
|
|
|
|
|
|
def _lazy_import(name):
|
|
module = importlib.import_module(f"whisperx.{name}")
|
|
return module
|
|
|
|
|
|
def load_align_model(*args, **kwargs):
|
|
alignment = _lazy_import("alignment")
|
|
return alignment.load_align_model(*args, **kwargs)
|
|
|
|
|
|
def align(*args, **kwargs):
|
|
alignment = _lazy_import("alignment")
|
|
return alignment.align(*args, **kwargs)
|
|
|
|
|
|
def load_model(*args, **kwargs):
|
|
asr = _lazy_import("asr")
|
|
return asr.load_model(*args, **kwargs)
|
|
|
|
|
|
def load_audio(*args, **kwargs):
|
|
audio = _lazy_import("audio")
|
|
return audio.load_audio(*args, **kwargs)
|
|
|
|
|
|
def assign_word_speakers(*args, **kwargs):
|
|
diarize = _lazy_import("diarize")
|
|
return diarize.assign_word_speakers(*args, **kwargs)
|