updated setup.py

setup.py

@@ -1,14 +1,21 @@
 from distutils.core import setup
+import setuptools
+
+with open('README.md') as readme_file: readme = readme_file.read()
+
 setup(
   name = 'ktrain',
-  packages = ['ktrain'],
-  version = '0.1.0',
+  #packages = ['ktrain'],
+  packages = setuptools.find_packages(),
+  version = '0.1.1',
   license='MIT',
   description = 'ktrain is a lightweight wrapper for Keras to help train neural networks',
+  long_description = readme,
+  long_description_content_type = 'text/markdown',
   author = 'Arun S. Maiya',
   author_email = '[email protected]',
   url = 'https://github.com/amaiya/ktrain',
-  download_url = 'https://github.com/amaiya/ktrain/archive/v0.1.0.tar.gz',
+  download_url = 'https://github.com/amaiya/ktrain/archive/v0.1.1.tar.gz',
   keywords = ['keras', 'deep learning', 'machine learning'],
   install_requires=[
           'keras >= 2.2.4',