38 lines
957 B
Python
38 lines
957 B
Python
|
import argparse, json, sys, time, random
|
||
|
|
import spacy
|
||
|
|
from aitextgen import aitextgen
|
||
|
|
|
||
|
|
def main() -> int:
|
||
|
|
|
||
|
|
p = argparse.ArgumentParser()
|
||
|
|
p.add_argument("-c", "--config", type=str, default="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']:
|
||
|
|
a = aitextgen(model_folder=v['model_dir'], tokenizer_file=v['tokeniser_file'])
|
||
|
|
voices.append({"name": v["name"].upper(), "a": a})
|
||
|
|
|
||
|
|
nbr_voices = len(voices)
|
||
|
|
current_voice = ""
|
||
|
|
for i in range(args.iterations):
|
||
|
|
rindex = random.randint(0, nbr_voices - 1)
|
||
|
|
v = voices[rindex]
|
||
|
|
if v['name'] != current_voice:
|
||
|
|
print("==========")
|
||
|
|
print(v['name'] + ":")
|
||
|
|
current_voice = v['name']
|
||
|
|
t = v['a'].generate_one().strip()
|
||
|
|
print(t)
|
||
|
|
|
||
|
|
time.sleep(1)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
sys.exit(main())
|