diff --git a/requirements.txt b/requirements.txt index d5c3683..9a54e79 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,4 +5,5 @@ kaggle >= 1.5.8 pandas >= 1.1.2 scikit-learn >= 0.23.2 datasets >= 1.0.2 -tqdm == 4.41.0 \ No newline at end of file +tqdm == 4.41.0 +sentencepiece >= 0.1.94 \ No newline at end of file diff --git a/setup.py b/setup.py index e11aea5..21bb938 100644 --- a/setup.py +++ b/setup.py @@ -7,7 +7,7 @@ setup( name="detoxify", - version="0.1.0", + version="0.1.1", description="A python library for detecting toxic comments", long_description=long_description, long_description_content_type="text/markdown", @@ -15,8 +15,9 @@ author_email="laura@unitary.ai", url="https://github.com/unitaryai/detoxify", install_requires=[ - "transformers", - "torch", + "transformers >= 3.2.0", + "torch >= 1.7.0", + "sentencepiece >= 0.1.94" ], packages=find_packages(exclude=("tests", "src")), classifiers=[