TensorFlowTTS/examples/cpptflite/demo/text2ids.py

import sys
import re

eng_pat = re.compile("[a-zA-Z]+")

if __name__ == "__main__":
    argvs = sys.argv

    if (len(argvs) != 3):
        print("usage: python3 {} mapper.json text".format(argvs[0]))
    else:
        from tensorflow_tts.inference import AutoProcessor
        mapper_json = argvs[1]
        processor = AutoProcessor.from_pretrained(pretrained_path=mapper_json)

        input_text = argvs[2]

        if eng_pat.match(input_text):
            input_ids = processor.text_to_sequence(input_text)
        else:
            input_ids = processor.text_to_sequence(input_text, inference=True)

        print(" ".join(str(i) for i in input_ids))