setup.py

from setuptools import find_packages, setup

from whisperkit._version import __version__

with open('README.md') as f:
    readme = f.read()

setup(
    name='whisperkit',
    version=__version__,
    url='https://github.com/argmaxinc/whisperkittools',
    description="Argmax WhisperKit Optimization Toolkit",
    long_description=readme,
    long_description_content_type='text/markdown',
    author='Argmax, Inc.',
    install_requires=[
        "argmaxtools",
        "transformers",
        "huggingface-hub",
        "evaluate",
        "datasets",
        "librosa",
        "soundfile",
        "jiwer",
        "tabulate",
        "mlx",
        "tiktoken",
        "openai"
    ],
    packages=find_packages(),
    entry_points={
        "console_scripts": [
            "whisperkit-generate-model=scripts.generate_model:cli",
            "whisperkit-evaluate-model=scripts.evaluate_model:cli",
            "whisperkit-generate-readme=scripts.generate_readme:cli",
        ],
    },
    classifiers=[
        "Development Status :: 4 - Beta",
        "Intended Audience :: Developers",
        "Operating System :: MacOS :: MacOS X",
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.9",
        "Programming Language :: Python :: 3.10",
        "Programming Language :: Python :: 3.11",
        "Topic :: Scientific/Engineering",
        "Topic :: Software Development",
    ],
)