diff --git a/requirements.txt b/requirements.txt
index f183b0d..e3c694e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -29,6 +29,6 @@ seaborn==0.10.0
 segmentation-models==1.0.1
 Shapely==1.7.0
 tensorboard==2.0.2
-tensorflow==2.0.0
+tensorflow==2.6.4
 tqdm==4.46.0
 urllib3==1.25.8
\ No newline at end of file