From 386292bafcd2a51113509f0e18a37cbc4d44aec3 Mon Sep 17 00:00:00 2001
From: Shawn-Shan <shansixioing@uchicago.edu>
Date: Wed, 19 May 2021 10:19:36 -0500
Subject: [PATCH] fix dependency issue with tensorflow

---
 setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index af4674d..64e5a50 100644
--- a/setup.py
+++ b/setup.py
@@ -76,8 +76,8 @@ setup_requires = []
 
 install_requires = [
     'numpy>=1.19.5',
-    'tensorflow>=2.0.0',
-    'keras>=2.3.1',
+    'tensorflow==2.4.1',
+    'keras==2.4.3',
     'mtcnn',
     'pillow>=7.0.0',
     'bleach>=2.1.0'