Please include requirements.txt

[robmarkcole]

As title. I think #7 may be due to changes in newer version of tensorflow

[btphan95]

Yes, it seems like the newer version of TensorFlow is not compatible with the Keras methods used in this repository.
Here are the versions of Keras and TF that worked for me:
Keras 2.2.4
TF 1.13.1