82 lines
2.6 KiB
Python
82 lines
2.6 KiB
Python
from setuptools import setup, find_packages
|
|
|
|
requirements = [
|
|
"ruamel.yaml",
|
|
"torch==1.4.0",
|
|
"torchvision==0.5.0",
|
|
"nemo_toolkit[asr] @ git+https://github.com/NVIDIA/NeMo.git@09e3ba4dfe333f86d6c5c1048e07210924294be9#egg=nemo_toolkit",
|
|
]
|
|
|
|
extra_requirements = {
|
|
"server": ["rpyc~=4.1.4", "tqdm~=4.39.0"],
|
|
"data": [
|
|
"google-cloud-texttospeech~=1.0.1",
|
|
"tqdm~=4.39.0",
|
|
"pydub~=0.24.0",
|
|
"scikit_learn~=0.22.1",
|
|
"pandas~=1.0.3",
|
|
"boto3~=1.12.35",
|
|
"ruamel.yaml==0.16.10",
|
|
"pymongo==3.10.1",
|
|
"librosa==0.7.2",
|
|
"numba==0.48",
|
|
"matplotlib==3.2.1",
|
|
"pandas==1.0.3",
|
|
"tabulate==0.8.7",
|
|
"natural==0.2.0",
|
|
"num2words==0.5.10",
|
|
"typer[all]==0.3.1",
|
|
"python-slugify==4.0.0",
|
|
"rpyc~=4.1.4",
|
|
"lenses @ git+https://github.com/ingolemo/python-lenses.git@b2a2a9aa5b61540992d70b2cf36008d0121e8948#egg=lenses",
|
|
],
|
|
"validation": [
|
|
"rpyc~=4.1.4",
|
|
"pymongo==3.10.1",
|
|
"typer[all]==0.1.1",
|
|
"tqdm~=4.39.0",
|
|
"librosa==0.7.2",
|
|
"matplotlib==3.2.1",
|
|
"pydub~=0.24.0",
|
|
"streamlit==0.58.0",
|
|
"natural==0.2.0",
|
|
"stringcase==1.2.0",
|
|
"google-cloud-speech~=1.3.1",
|
|
]
|
|
# "train": [
|
|
# "torchaudio==0.5.0",
|
|
# "torch-stft==0.1.4",
|
|
# ]
|
|
}
|
|
packages = find_packages()
|
|
|
|
setup(
|
|
name="jasper-asr",
|
|
version="0.1",
|
|
description="Tool to get gcp alignments of tts-data",
|
|
url="http://github.com/malarinv/jasper-asr",
|
|
author="Malar Kannan",
|
|
author_email="malarkannan.invention@gmail.com",
|
|
license="MIT",
|
|
install_requires=requirements,
|
|
extras_require=extra_requirements,
|
|
packages=packages,
|
|
entry_points={
|
|
"console_scripts": [
|
|
"jasper_transcribe = jasper.transcribe:main",
|
|
"jasper_server = jasper.server:main",
|
|
"jasper_trainer = jasper.training.cli:main",
|
|
"jasper_evaluator = jasper.evaluate:main",
|
|
"jasper_data_tts_generate = jasper.data.tts_generator:main",
|
|
"jasper_data_conv_generate = jasper.data.conv_generator:main",
|
|
"jasper_data_nlu_generate = jasper.data.nlu_generator:main",
|
|
"jasper_data_rastrik_recycle = jasper.data.rastrik_recycler:main",
|
|
"jasper_data_server = jasper.data.server:main",
|
|
"jasper_data_validation = jasper.data.validation.process:main",
|
|
"jasper_data_preprocess = jasper.data.process:main",
|
|
"jasper_data_slu_evaluate = jasper.data.slu_evaluator:main",
|
|
]
|
|
},
|
|
zip_safe=False,
|
|
)
|