Cool! I also did something like this: https://github.com/hallvardnmbu/transformer

But on various data (i.e., separate model per source): the Bible, Don Quixote and Franz Kafka. (As well as a (bad!) lyrics generator, and translator.)