File size: 969 Bytes
5806e12 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
from setuptools import setup, find_packages
setup(
name="wtpsplit",
version="2.1.6",
packages=find_packages(),
description="Universal Robust, Efficient and Adaptable Sentence Segmentation",
author="Markus Frohmann, Igor Sterner, Benjamin Minixhofer",
author_email="[email protected]",
install_requires=[
# "onnxruntime>=1.13.1", # can make conflicts between onnxruntime and onnxruntime-gpu
"transformers>=4.22.2",
"huggingface-hub",
"numpy>=1.0",
"scikit-learn>=1",
"tqdm",
"skops",
"pandas>=1",
"cached_property", # for Py37
"mosestokenizer",
"adapters>=1.0.1",
],
extras_require={
'onnx-gpu': ['onnxruntime-gpu>=1.13.1'],
'onnx-cpu': ['onnxruntime>=1.13.1'],
},
url="https://github.com/segment-any-text/wtpsplit",
package_data={"wtpsplit": ["data/*"]},
include_package_data=True,
license="MIT",
)
|