diff --git a/setup.py b/setup.py
index 7c0505de50afe8fa10a4bbdd67799b3a03f3b498..fa8c1cdd9fcb7985008fa1faede9fc5298ae38ae 100644
--- a/setup.py
+++ b/setup.py
@@ -13,5 +13,5 @@ setup(
     description=metainfo['title'],
     long_description=metainfo['description'],
     packages=find_packages(),
-    install_requires=['tensorflow==1.13.1', 'keras==2.2.4', 'keras-vis', 'numpy==1.16.4', 'scipy==1.1.0', 'matplotlib', 'pandas', 'seaborn', 'pymatgen==2020.3.13', 'sklearn'],
+    install_requires=['tensorflow==1.13.1', 'keras==2.2.4', 'numpy==1.16.4', 'scipy==1.1.0', 'matplotlib', 'pandas', 'seaborn', 'pymatgen==2020.3.13', 'sklearn'],
 )