forked from utterworks/fast-bert
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
26 lines (25 loc) · 985 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
from io import open
from setuptools import setup, find_packages
setup(name='fast_bert',
version='0.1.2',
description='AI Library using BERT',
author='Kaushal Trivedi',
author_email='kaushaltrivedi@me.com',
license='Apache2',
url='https://github.com/kaushaltrivedi/fast-bert',
long_description=open("README.md", "r", encoding='utf-8').read(),
long_description_content_type="text/markdown",
keywords='BERT NLP deep learning google',
packages=find_packages(exclude=["*.tests", "*.tests.*",
"tests.*", "tests"]),
install_requires=[
'pytorch-pretrained-bert',
'fastai'
],
classifiers=[
'Intended Audience :: Science/Research',
'License :: OSI Approved :: Apache Software License',
'Programming Language :: Python :: 3',
'Topic :: Scientific/Engineering :: Artificial Intelligence',
],
zip_safe=False)