File size: 657 Bytes
d5ee97c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
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)) |