update install

.gitignore

@@ -4,6 +4,7 @@ basetts_outputs_use_bert/
 basetts_outputs/
 multilingual_ckpts
 basetts_outputs_package/
+build/
 *.egg-info/
 
 *.zip

README.md

@@ -1,5 +1,2 @@
-##  Usage (After installing OpenVoice)
+# MyShell TTS Base Speakers
-```
+
-pip install .
-python -m unidic download
-```

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,
+    },
 )