diff --git a/setup.py b/setup.py index 7e7470d..8e798c8 100644 --- a/setup.py +++ b/setup.py @@ -1,12 +1,31 @@ from setuptools import setup, find_packages +with open("README.md", "r", encoding="utf-8") as fh: + long_description = fh.read() + setup( name='stable-audio-tools', - version='0.0.16', + version='0.1.0', url='https://github.com/Stability-AI/stable-audio-tools.git', author='Stability AI', + author_email='info@stability.ai', description='Training and inference tools for generative audio models from Stability AI', - packages=find_packages(), + long_description=long_description, + long_description_content_type="text/markdown", + packages=find_packages(), + include_package_data=True, + classifiers=[ + "Development Status :: 3 - Alpha", + "Intended Audience :: Developers", + "Intended Audience :: Science/Research", + "License :: OSI Approved :: MIT License", + "Programming Language :: Python :: 3", + "Programming Language :: Python :: 3.7", + "Programming Language :: Python :: 3.8", + "Programming Language :: Python :: 3.9", + "Topic :: Scientific/Engineering :: Artificial Intelligence", + ], + python_requires='>=3.7', install_requires=[ 'aeiou==0.0.20', 'alias-free-torch==0.0.6', @@ -25,7 +44,7 @@ 'pandas==2.0.2', 'pedalboard==0.7.4', 'prefigure==0.0.9', - 'pytorch_lightning==2.1.0', + 'pytorch_lightning==2.1.0', 'PyWavelets==1.4.1', 'safetensors', 'sentencepiece==0.1.99', @@ -41,4 +60,9 @@ 'webdataset==0.2.48', 'x-transformers<1.27.0' ], -) \ No newline at end of file + entry_points={ + 'console_scripts': [ + 'stable-audio-tools=stable_audio_tools.cli:main', + ], + }, +)