"--tokenizer",
type=str,
default="tokenizer.pickle",
help="Path to tokenizer file to load.")
args = parser.parse_args()
return vars(args)
if __name__ == "__main__":
args = parse_args()
if not args["load"] and "corpus" not in args:
raise ValueError("Need a corpus for the parser to parse.")
bot = Processor()
if args["load"]:
bot.load_all(args["encoder"], args["decoder"], args["tokenizer"])
else:
p = Parser(args["corpus"])
q, a = p.main(args["google"], args["qa_save"])
bot = Processor()
bot.main(q, a)
if args["model_save"]:
bot.save_model(bot.encoder, name="google_enc.h5")
bot.save_model(bot.decoder, name="google_dec.h5")
bot.save_tokenizer(bot.tokenizer, name="google.token.pickle")
if args["tts"]:
tts = TTS()
stt = STT()
