From 5dec83433324b10a5bf4cf0290e3fd8aae2e1e84 Mon Sep 17 00:00:00 2001
From: jessevig <jesse.vig@gmail.com>
Date: Thu, 6 May 2021 04:12:20 -0700
Subject: [PATCH] Add setup.py. Minor README change.

---
 README.md |  4 +++-
 setup.py  | 23 +++++++++++++++++++++++
 2 files changed, 26 insertions(+), 1 deletion(-)
 create mode 100644 setup.py

diff --git a/README.md b/README.md
index 316a2d5..2868675 100644
--- a/README.md
+++ b/README.md
@@ -191,6 +191,8 @@ model_view(
 )
 ```
 
+You may select `Encoder`, `Decoder`, or `Cross` attention from the drop-down in the upper left corner.
+
 ### Running a sample notebook
 
 ```
@@ -220,7 +222,7 @@ head_view(attention, tokens, layer=2, heads=[3,5])
 
 You may also pre-select a specific `layer` and single `head` for the neuron view.
 
-#### Dark/light mode
+#### Dark / light mode
 
 The model view and neuron view support dark (default) and light modes. You may turn off dark mode in these views using
 the `display_mode` parameter:
diff --git a/setup.py b/setup.py
new file mode 100644
index 0000000..325998d
--- /dev/null
+++ b/setup.py
@@ -0,0 +1,23 @@
+import pathlib
+from setuptools import setup
+
+# The directory containing this file
+HERE = pathlib.Path(__file__).parent
+
+# The text of the README file
+README = (HERE / "README.md").read_text()
+
+# This call to setup() does all the work
+setup(
+    name="bertviz",
+    version="1.1.0",
+    description="Attention visualization tool for NLP Transformer models.",
+    long_description=README,
+    long_description_content_type="text/markdown",
+    url="https://github.com/jessevig/bertviz",
+    author="Jesse Vig",
+    license="Apache 2.0",
+    packages=["bertviz"],
+    include_package_data=True,
+    install_requires=["transformers>=2.0", "torch>=1.0", "tqdm", "boto3", "requests", "regex", "sentencepiece"],
+)
\ No newline at end of file