Bare Bones Neural Net

gen.py

#! /usr/bin/env python3

import pathlib
import sys
from textgenrnn import textgenrnn
stem = sys.argv[1]
count = int(sys.argv[2])
hdfile = f"{stem}.hdf5"
path = pathlib.Path(hdfile)

if path.exists():
    t = textgenrnn(hdfile)
else:
    t = textgenrnn()
    t.train_from_file(f"{stem}.txt", num_epochs=5)
t.generate(count, temperature=0.5)