forked from ramsrigouthamg/Questgen.ai
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
33 lines (31 loc) · 1 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
from distutils.core import setup
setup(name='Questgen',
version='1.0.0',
description='Question generator from any text',
author='Questgen contributors',
author_email='[email protected]',
packages=['Questgen', 'Questgen.encoding', 'Questgen.mcq'],
url="https://github.com/ramsrigouthamg/Questgen.ai",
install_requires=[
'torch==1.9.0',
'transformers==3.0.2',
'pytorch_lightning==0.8.1',
'sense2vec==1.0.3',
'strsim==0.0.3',
'six==1.15.0',
'networkx==2.4.0',
'numpy',
'scipy',
'scikit-learn',
'unidecode==1.1.1',
'future==0.18.2',
'joblib==0.14.1',
'spacy==2.2.4',
'pytz==2020.1',
'python-dateutil==2.8.1',
'boto3==1.14.40',
'flashtext==2.7',
'pandas'
],
package_data={'Questgen': ['questgen.py', 'mcq.py', 'train_gpu.py', 'encoding.py']}
)