forked from vsuthichai/paraphraser
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
68 lines (57 loc) · 1.95 KB
/
setup.py
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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
from setuptools import setup, find_packages
from codecs import open
from os import path
from setuptools.command.install import install
here = path.abspath(path.dirname(__file__))
with open(path.join(here, 'README.md'), encoding='utf-8') as f:
long_desc = f.read()
class DownloadCorpora(install):
def run(self):
install.run(self)
import spacy
import nltk
nltk.download('wordnet')
spacy.cli.download('download', 'en')
class DownloadParaphraseModel(install):
def run(self):
install.run(self)
from paraphaser.download_models import download_file_from_google_drive
download_file_from_google_drive('19QDCd4UMgt3FtlYYwu0qZU3G1F9_XCvk',
'paraphrase-model.tar.gz')
setup(
name='paraphraser',
version='0.1.0',
description='Generate sentence paraphrases given an input sentence',
long_description=long_desc,
url='https://github.com/vsuthichai/paraphraser',
author='Victor Suthichai',
author_email='[email protected]',
# https://pypi.python.org/pypi?%3Aaction=list_classifiers
classifiers=[
'Development Status :: 3 - Alpha',
'Intended Audience :: Developers',
'Programming Language :: Python :: 2.7',
'Programming Language :: Python :: 3',
'Programming Language :: Python :: 3.4',
'Programming Language :: Python :: 3.5',
'Programming Language :: Python :: 3.6'
],
keywords=[
'paraphraser'
],
py_modules=['paraphraser.synonym_model', 'paraphraser.inference', 'paraphraser.download_models'],
#packages=find_packages(exclude=['contrib', 'docs', 'tests']),
#install_requires=['nltk', 'spacy', 'ipython'],
install_requires=[],
extras_require={
},
package_data={
},
data_files=[],
entry_points={
},
cmdclass={
'download_model': DownloadParaphraseModel
#'download_corpora': DownloadCorpora
}
)