mirror of
https://github.com/m-bain/whisperX.git
synced 2025-07-01 18:17:27 -04:00
38 lines
1.0 KiB
Python
38 lines
1.0 KiB
Python
import os
|
|
import platform
|
|
import pkg_resources
|
|
from setuptools import setup, find_packages
|
|
|
|
|
|
def get_pyannote_audio_version():
|
|
machine = platform.machine()
|
|
system = platform.system()
|
|
version = "3.0.0" if machine == "aarch64" or system == "Darwin" else "3.0.1"
|
|
return version
|
|
|
|
|
|
setup(
|
|
name="whisperx",
|
|
py_modules=["whisperx"],
|
|
version="3.1.1",
|
|
description="Time-Accurate Automatic Speech Recognition using Whisper.",
|
|
readme="README.md",
|
|
python_requires=">=3.8",
|
|
author="Max Bain",
|
|
url="https://github.com/m-bain/whisperx",
|
|
license="MIT",
|
|
packages=find_packages(exclude=["tests*"]),
|
|
install_requires=[
|
|
str(r)
|
|
for r in pkg_resources.parse_requirements(
|
|
open(os.path.join(os.path.dirname(__file__), "requirements.txt"))
|
|
)
|
|
]
|
|
+ [f"pyannote.audio=={get_pyannote_audio_version()}"],
|
|
entry_points={
|
|
"console_scripts": ["whisperx=whisperx.transcribe:cli"],
|
|
},
|
|
include_package_data=True,
|
|
extras_require={"dev": ["pytest"]},
|
|
)
|