fix dependency issue with tensorflow

2021-05-19 10:19:36 -05:00 · 2021-05-19 10:19:36 -05:00 · 386292bafc
commit 386292bafc
--- 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'