forked from ramsrigouthamg/Questgen.ai
-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
30 lines (29 loc) · 1003 Bytes
/
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
from distutils.core import setup
setup(name='Questgen',
version='1.0.1',
description='Question generator from any text',
author='Questgen contributors',
author_email='vaibhavtiwarifu@gmail.com',
packages=['Questgen', 'Questgen.encoding', 'Questgen.mcq'],
url="https://github.com/ramsrigouthamg/Questgen.ai",
install_requires=[
'torch==2.0.1',
'transformers==4.29.2',
'sense2vec==2.0.2',
'strsim==0.0.3',
'six==1.16.0',
'networkx==3.1',
'numpy==1.22.4',
'scipy==1.10.1',
'scikit-learn==1.2.2',
'unidecode==1.3',
'future==0.18.3',
'joblib==1.2.0',
'pytz==2022.7.1',
'python-dateutil==2.8.2',
'flashtext==2.7',
'pandas==1.5.3',
'sentencepiece==0.1.99'
],
package_data={'Questgen': ['questgen.py', 'mcq.py', 'train_gpu.py', 'encoding.py']}
)