setup.py

#!/usr/bin/env python
from setuptools import setup
from setuptools import find_packages

setup(
    name='text-generator',
    version='0.0.0',
    packages=find_packages(exclude=["*_tests"]),
    license='MIT license',
    long_description=open('README.md').read(),
    long_description_content_type='text/markdown',
    entry_points = {
        'console_scripts': [
            'text_generator = text_generator.cli:cli',
        ],
    },
    install_requires = [
        'ariadne',
        'click',
        'flask',
        'torch',
        'transformers'
    ],
    extras_require={
        'dev': [
            'pylint',
            'coverage',
            'twine'
        ]
    },
    classifiers=[
        "Programming Language :: Python :: 2.7",
        "Programming Language :: Python :: 3.6",
        "License :: OSI Approved :: MIT License",
        "Operating System :: OS Independent",
        "Intended Audience :: Developers",
        "Environment :: Console"
    ]
)