diff --git a/setup.py b/setup.py
index ef4f4b7170ce50983900929788383724fb4e81cc..de85809933a517d13f09d6f0622f0657fb1a94b1 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'],
 )