diff --git a/apis/train_model/requirements.txt b/apis/train_model/requirements.txt
index 137635b..ae98116 100644
--- a/apis/train_model/requirements.txt
+++ b/apis/train_model/requirements.txt
@@ -8,4 +8,4 @@ Pillow==6.2.0
 lime==0.2.0.0
 matplotlib==3.1.1 
 autokeras==1.0.2
-tensorflow==2.1.0
\ No newline at end of file
+tensorflow==2.5.3
\ No newline at end of file