setup.py

app.py

@@ -0,0 +1,46 @@
+import io
+import os
+import re
+from os import path
+
+from setuptools import find_packages
+from setuptools import setup
+
+
+this_directory = path.abspath(path.dirname(__file__))
+with open(path.join(this_directory, 'README.md'), encoding='utf-8') as f:
+    long_description = f.read()
+
+
+setup(
+    name="mario-gpt",
+    version="0.1.3",
+    url="https://github.com/shyamsn97/mario-gpt",
+    license='MIT',
+
+    author="Shyam Sudhakaran",
+    author_email="[email protected]",
+
+    description="Generating Mario Levels with GPT2. Code for the paper: 'MarioGPT: Open-Ended Text2Level Generation through Large Language Models', https://arxiv.org/abs/2302.05981",
+
+    long_description=long_description,
+    long_description_content_type="text/markdown",
+
+    include_package_data = True,
+
+    packages=find_packages(exclude=('tests',)),
+
+    install_requires=[
+        'torch',
+        'transformers',
+        'scipy',
+        'tqdm',
+        'pillow',
+    ],
+
+    classifiers=[
+        'Development Status :: 2 - Pre-Alpha',
+        'License :: OSI Approved :: MIT License',
+        'Programming Language :: Python :: 3',
+    ],
+)