diff options
| author | Maarten Grootendorst <maarten_grootendorst@hotmail.com> | 2021-06-30 12:47:45 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-06-30 12:47:45 +0300 |
| commit | 25dab3a763e5cc0450346dcf061754f71f184ae3 (patch) | |
| tree | 6e734d3d57a732a3bc213cc45dd16707b71bf04a /setup.py | |
| parent | eb6d0865c958a474f3554518539c7a37dbd9856b (diff) | |
v0.4 (#43)v0.4.0
* Use paraphrase-MiniLM-L6-v2 as the default embedding model
* Highlight a document's keywords
* Added FAQ
Diffstat (limited to 'setup.py')
| -rw-r--r-- | setup.py | 4 |
1 files changed, 3 insertions, 1 deletions
@@ -9,6 +9,7 @@ base_packages = [ "sentence-transformers>=0.3.8", "scikit-learn>=0.22.2", "numpy>=1.18.5", + "rich>=10.4.0" ] docs_packages = [ @@ -18,6 +19,7 @@ docs_packages = [ ] flair_packages = [ + "transformers==3.5.1", "torch>=1.4.0,<1.7.1", "flair==0.7" ] @@ -46,7 +48,7 @@ with open("README.md", "r", encoding='utf-8') as fh: setup( name="keybert", packages=find_packages(exclude=["notebooks", "docs"]), - version="0.3.0", + version="0.4.0", author="Maarten Grootendorst", author_email="maartengrootendorst@gmail.com", description="KeyBERT performs keyword extraction with state-of-the-art transformer models.", |