diff --git a/setup.py b/setup.py index 830b50f8..ccf7f51e 100644 --- a/setup.py +++ b/setup.py @@ -15,8 +15,8 @@ url="https://github.com/Palashio/libra", install_requires=['colorama', 'transformers==2.11.0', - 'tensorflow', - 'keras', + 'tensorflow==2.2.0', + 'keras==2.4.3', 'numpy', 'pandas', 'sklearn',