diff --git a/requirements.txt b/requirements.txt
index 6382a53..a182e3c 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,5 +3,5 @@ numpy
 pillow
 scikit-image
 scipy
-tensorflow==1.15.2
+tensorflow==2.6.4
 pyyaml