update install

setup.py

@@ -1,9 +1,24 @@
 import os 
 from setuptools import setup, find_packages
+from setuptools.command.develop import develop
+from setuptools.command.install import install
 
 cwd = os.path.dirname(os.path.abspath(__file__))
 requirements = open(os.path.join(cwd, "requirements.txt"), "r").readlines()
 
+class PostInstallCommand(install):
+    """Post-installation for installation mode."""
+    def run(self):
+        install.run(self)
+        os.system('python -m unidic download')
+
+
+class PostDevelopCommand(develop):
+    """Post-installation for development mode."""
+    def run(self):
+        develop.run(self)
+        os.system('python -m unidic download')
+
 setup(
     name='MyShellTTSBase',
     version='0.1.0',
@@ -13,4 +28,8 @@ setup(
     package_data={
         '': ['*.txt', 'cmudict_*'],
     },
+    cmdclass={
+        'develop': PostDevelopCommand,
+        'install': PostInstallCommand,
+    },
 )