nix_tts_simple/tts.py

import pickle
import os

from .tokenizer_lojban import NixTokenizerEN
from .inference import NixTTSInference

def generate_voice(
    input_text,
    model_dir
):
    tokenizer = NixTokenizerEN(pickle.load(open(os.path.join(model_dir, "tokenizer_state.pkl"), "rb")))
    c, c_length, phoneme = tokenizer([input_text])
    inference = NixTTSInference(model_dir)
    voice = inference.vocalize(c, c_length)

    return [input_text, (22050, voice[0,0])]