41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
import argparse, json, sys, time, random
|
|
import utterance.voice
|
|
import utterance.utils
|
|
|
|
UTTERANCE_LEN = 64 #<--------------- these should be in config
|
|
|
|
def main() -> int:
|
|
|
|
p = argparse.ArgumentParser()
|
|
p.add_argument("-c", "--config", type=str, default="voice.config.json", help="configuratin file")
|
|
p.add_argument("-i", "--iterations", type=int, default=10, help="number of iterations")
|
|
args = p.parse_args()
|
|
|
|
print(args)
|
|
|
|
with open(args.config) as f:
|
|
conf = json.load(f)
|
|
|
|
voices = []
|
|
for v in conf['voices']:
|
|
model = v['model']
|
|
voice = utterance.voice.Voice(name=v["name"].upper(), model=model['model_dir'], tokenizer=model['tokeniser_file'], temp=float(model["temperature"]), length=UTTERANCE_LEN)
|
|
voices.append(voice)
|
|
|
|
current_voice_name = ""
|
|
for i in range(args.iterations):
|
|
v = random.sample(voices, 1)[0]
|
|
if v.name != current_voice_name:
|
|
current_voice_name = v.name
|
|
print("==========")
|
|
print(v.name + ":")
|
|
t = v.utter_one()
|
|
t = utterance.utils.clean(t)
|
|
t = utterance.utils.format(t)
|
|
print(t)
|
|
|
|
# time.sleep(1)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main()) |