diff --git a/README.md b/README.md
index a6916e5..43934ac 100644
--- a/README.md
+++ b/README.md
@@ -24,6 +24,7 @@ If you want to run the code yourself, you can clone this repo and download the m
 ### Dependencies (for running the code yourself)
 - Python 3
 - TensorFlow 1.10
+- Keras 2.2.4
 - Pillow
 - h5py
 
diff --git a/requirements.txt b/requirements.txt
index e60d2ba..cfa1f0f 100644
Binary files a/requirements.txt and b/requirements.txt differ