buzz/tests/model_loader.py
2023-04-28 22:28:05 +01:00

18 lines
537 B
Python

from buzz.model_loader import TranscriptionModel, get_local_model_path, ModelDownloader
def get_model_path(transcription_model: TranscriptionModel) -> str:
path = get_local_model_path(model=transcription_model)
if path is not None:
return path
model_loader = ModelDownloader(model=transcription_model)
model_path = ''
def on_load_model(path: str):
nonlocal model_path
model_path = path
model_loader.signals.finished.connect(on_load_model)
model_loader.run()
return model_path