diff --git a/pyproject.toml b/pyproject.toml index 10d593f..b37c972 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,11 +1,11 @@ [metadata] name = 'zeroshot_topics' version = '0.0.1' -description = '' +description = 'Topic Inference with Zeroshot models' author = 'AnjanaRita' author_email = 'ritaanjana1993@gmail.com' license = 'MIT/Apache-2.0' -url = 'https://github.com/_/zeroshot_topics' +url = 'https://github.com/AnjanaRita/zeroshot_topics' [requires] python_version = ['2.7', '3.5', '3.6', 'pypy', 'pypy3'] diff --git a/requirements.txt b/requirements.txt index d6e1198..9cf8170 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1 +1,4 @@ --e . +attrs==20.3.0 +nltk==3.6.5 +keybert==0.5.0 +transformers==4.11.0 \ No newline at end of file diff --git a/setup.py b/setup.py index 8b9e817..39df15a 100644 --- a/setup.py +++ b/setup.py @@ -18,12 +18,12 @@ with open('zeroshot_topics/__init__.py', 'r') as f: with open('README.rst', 'r', encoding='utf-8') as f: readme = f.read() -REQUIRES = [] +REQUIRES = [i.strip() for i in open("requirements.txt").readlines()] kwargs = { 'name': 'zeroshot_topics', 'version': version, - 'description': '', + 'description': 'Topic Inference with Zeroshot models', 'long_description': readme, 'author': 'AnjanaRita', 'author_email': 'ritaanjana1993@gmail.com', diff --git a/zeroshot_topics/__init__.py b/zeroshot_topics/__init__.py index b8023d8..a7b9a32 100644 --- a/zeroshot_topics/__init__.py +++ b/zeroshot_topics/__init__.py @@ -1 +1,3 @@ __version__ = '0.0.1' + +from .zeroshot_tm import ZeroShotTopicFinder