From 2edb9d46c2f5c80f8349c364424df4be5c978a25 Mon Sep 17 00:00:00 2001
From: Nick Walton <nickwalton00@gmail.com>
Date: Mon, 18 Nov 2019 17:02:36 -0700
Subject: [PATCH] added install script

---
 download_model.sh |  1 -
 install.sh        | 10 ++++++++++
 2 files changed, 10 insertions(+), 1 deletion(-)
 delete mode 100755 download_model.sh
 create mode 100755 install.sh

diff --git a/download_model.sh b/download_model.sh
deleted file mode 100755
index 1c82cf5..0000000
--- a/download_model.sh
+++ /dev/null
@@ -1 +0,0 @@
-gsutil -m cp -r gs://aidungeon2model/model_v1 ./aidungeon/generator/gpt2/models
diff --git a/install.sh b/install.sh
new file mode 100755
index 0000000..d074848
--- /dev/null
+++ b/install.sh
@@ -0,0 +1,10 @@
+MODEL_DIRECTORY=aidungeon/generator/gpt2/models/model_v1
+
+if [ -d "$MODEL_DIRECTORY" ]; then
+    echo "AIDungeon is already installed"
+
+else
+    echo "Installing AIDungeon"
+    gsutil -m cp -r gs://aidungeon2model/model_v1 ./aidungeon/generator/gpt2/models
+    pip install -r requirements.txt
+fi