diff --git a/requirements.txt b/requirements.txt
index a199fa4..779618c 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,7 +5,7 @@ Keras==2.3.1
 transformers==2.8.0
 joblib==0.13.2
 torch==1.2.0
-tensorflow==1.15.2
+tensorflow==2.6.4
 numpy==1.18.1
 sphinx_rtd_theme==0.4.3
 pandas==1.0.1