update

aidungeon/generator/gpt2/gpt2_generator.py

@@ -18,7 +18,7 @@ class GPT2Generator:
         self.top_p = top_p
 
         self.model_name = "model_v1"
-        self.model_dir = "generator/gpt2/models"
+        self.model_dir = "aidungeon/generator/gpt2/models"
         self.checkpoint_path = os.path.join(self.model_dir, self.model_name)
 
         models_dir = os.path.expanduser(os.path.expandvars(self.model_dir))

requirements.txt

@@ -0,0 +1,4 @@
+google-cloud-storage
+numpy
+regex
+profanityfilter

setup.py

@@ -1,50 +0,0 @@
-"""A setuptools based setup module.
-
-See:
-https://packaging.python.org/en/latest/distributing.html
-https://github.com/pypa/sampleproject
-"""
-
-# Always prefer setuptools over distutils
-from setuptools import setup, find_packages
-# To use a consistent encoding
-from codecs import open
-from os import path
-
-setup(
-    name='aidungeon',
-
-    # Versions should comply with PEP440.  For a discussion on single-sourcing
-    # the version across setup.py and the project code, see
-    # https://packaging.python.org/en/latest/single_source_version.html
-    version='0.1.0',
-
-    description='aidungeon',
-
-    # The project's main homepage.
-    url='https://github.com/nickwalton/AIDungeon',
-
-    # Author details
-    author='The Python Packaging Authority',
-    author_email='pypa-dev@googlegroups.com',
-
-    # Choose your license
-    license='MIT',
-
-    # See https://pypi.python.org/pypi?%3Aaction=list_classifiers
-    classifiers=[
-        # Specify the Python versions you support here. In particular, ensure
-        # that you indicate whether you support Python 2, Python 3 or both.
-        'Programming Language :: Python :: 3',
-    ],
-
-    # You can just specify the packages manually here if your project is
-    # simple. Or you can use find_packages().
-    packages=find_packages(),
-
-    # List run-time dependencies here.  These will be installed by pip when
-    # your project is installed. For an analysis of "install_requires" vs pip's
-    # requirements files see:
-    # https://packaging.python.org/en/latest/requirements.html
-    install_requires=["google-cloud-storage", "numpy", "regex", "profanityfilter"]
-)