diff --git a/flowers/requirements.txt b/flowers/requirements.txt
index a45cc69bd..78fe3b39d 100644
--- a/flowers/requirements.txt
+++ b/flowers/requirements.txt
@@ -1,3 +1,3 @@
 apache-beam[gcp]==2.8.0
 pillow==4.0.0
-tensorflow==1.10
+tensorflow==2.5.3
diff --git a/flowers/setup.py b/flowers/setup.py
index 163e0c6eb..7c41eb66e 100644
--- a/flowers/setup.py
+++ b/flowers/setup.py
@@ -16,7 +16,7 @@
 from setuptools import setup
 
 REQUIRED_PACKAGES = [
-  'tensorflow==1.0.1',
+  'tensorflow==2.5.3',
 ]
 
 setup(