diff --git a/interpretability/requirements.txt b/interpretability/requirements.txt
index 7a9fef2..424f8a2 100644
--- a/interpretability/requirements.txt
+++ b/interpretability/requirements.txt
@@ -4,7 +4,7 @@ seaborn
 h5py
 keras==2.1.3
 scikit-learn
-tensorflow==1.6
+tensorflow==2.5.3
 pillow
 scipy
 matplotlib