added sentencepiece to requirements as it no longer comes with transf…

…ormers
unitaryai · Dec 9, 2020 · 1da4222 · 1da4222
1 parent c8e66f8
commit 1da4222
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git 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
+tqdm == 4.41.0
+sentencepiece >= 0.1.94
diff --git a/setup.py b/setup.py
@@ -7,16 +7,17 @@
 
 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",
     author="Unitary",
     author_email="[email protected]",
     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=[