from models.nllb import nllb
#from models.parakeet import parakeet_ctc_model
model_nllb, tokenizer_nllb = nllb()

from models.TTS_utils import load_manual_xtts_v2


config_path = "test/config.json"
model_path = "test"

xtts_v2_model = load_manual_xtts_v2(config_path, model_path)


def get_nllb_model_and_tokenizer():
    return model_nllb, tokenizer_nllb

def get_xtts_model():
    return xtts_v2_model