nahue-passano's picture
Update setup.py
fa240af
import os
from setuptools import setup, find_packages
install_requires = open(os.path.join(os.path.dirname(__file__), "requirements.txt")).readlines()
version = None
license = None
with open(os.path.join(os.path.dirname(__file__), "whisper_timestamped", "transcribe.py")) as f:
for line in f:
if line.strip().startswith("__version__"):
version = line.split("=")[1].strip().strip("\"'")
if version and license:
break
if line.strip().startswith("__license__"):
license = line.split("=")[1].strip().strip("\"'")
if version and license:
break
assert version and license
setup(
name="whisper-timestamped",
py_modules=["whisper_timestamped"],
version=version,
description="Add to OpenAI Whisper the capability to give word timestamps",
python_requires=">=3.7",
author="Jeronymous",
url="https://github.com/linto-ai/whisper-timestamped",
license=license,
packages=find_packages(exclude=["tests*"]),
install_requires=install_requires,
entry_points = {
'console_scripts': [
'whisper_timestamped=whisper_timestamped.transcribe:cli',
'whisper_timestamped_make_subtitles=whisper_timestamped.make_subtitles:cli'
],
},
include_package_data=True,
extras_require={
'dev': ['matplotlib', 'jsonschema', 'transformers'],
'vad': ['onnxruntime', 'torchaudio'],
},
)