diff --git a/setup.py b/setup.py index d7ced25..9f7ee68 100644 --- a/setup.py +++ b/setup.py @@ -2,13 +2,20 @@ import os from setuptools import find_packages, setup +base_dir = os.path.dirname(os.path.abspath(__file__)) + + +def get_long_description(): + readme_path = os.path.join(base_dir, "README.md") + with open(readme_path, encoding="utf-8") as readme_file: + return readme_file.read() + def get_requirements(path): with open(path, encoding="utf-8") as requirements: return [requirement.strip() for requirement in requirements] -base_dir = os.path.dirname(os.path.abspath(__file__)) install_requires = get_requirements(os.path.join(base_dir, "requirements.txt")) conversion_requires = get_requirements( os.path.join(base_dir, "requirements.conversion.txt") @@ -17,8 +24,26 @@ conversion_requires = get_requirements( setup( name="faster-whisper", version="0.1.0", + license="MIT", description="Faster Whisper transcription with CTranslate2", + long_description=get_long_description(), + long_description_content_type="text/markdown", author="Guillaume Klein", + url="https://github.com/guillaumekln/faster-whisper", + classifiers=[ + "Development Status :: 4 - Beta", + "Intended Audience :: Developers", + "Intended Audience :: Science/Research", + "License :: OSI Approved :: MIT License", + "Programming Language :: Python :: 3", + "Programming Language :: Python :: 3 :: Only", + "Programming Language :: Python :: 3.8", + "Programming Language :: Python :: 3.9", + "Programming Language :: Python :: 3.10", + "Programming Language :: Python :: 3.11", + "Topic :: Scientific/Engineering :: Artificial Intelligence", + ], + keywords="openai whisper speech ctranslate2 inference quantization transformer", python_requires=">=3.8", install_requires=install_requires, extras_require={