diff --git a/interpretability/requirements.txt b/interpretability/requirements.txt
index 7a9fef2..8c5b164 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.6.4
 pillow
 scipy
 matplotlib