diff --git a/setup.py b/setup.py
index 18670c0..7190a76 100644
--- a/setup.py
+++ b/setup.py
@@ -11,6 +11,7 @@ setup(
         "soundfile==0.13.1",
         "basic-pitch==0.4.0",
         "music21==9.9.1",
+        "tf-keras==2.15.0",
     ],
     classifiers=[
         "Programming Language :: Python :: 3",