diff --git a/.dockerignore b/.dockerignore
new file mode 100644
index 0000000..3fa9317
--- /dev/null
+++ b/.dockerignore
@@ -0,0 +1,5 @@
+Dockerfile
+README.md
+*.pyc
+*.pyo
+__pycache__
\ No newline at end of file
diff --git a/Dockerfile b/Dockerfile
new file mode 100644
index 0000000..3969100
--- /dev/null
+++ b/Dockerfile
@@ -0,0 +1,12 @@
+FROM nvcr.io/nvidia/tensorflow:19.08-py3
+
+ENV APP_HOME /app
+WORKDIR $APP_HOME
+COPY . ./
+
+RUN pip install -r requirements.txt
+RUN cd generator/ctrl
+RUN ./install_ctrl_py3.sh
+RUN ./download_model.sh
+
+CMD exec guincorn --bind :%PORT --workers 1 --threads 8 app:app
diff --git a/console_play.py b/console_play.py
index f11a3c1..4e05ad1 100644
--- a/console_play.py
+++ b/console_play.py
@@ -25,15 +25,18 @@ def play_unconstrained():
     #generator = WebGenerator(CRED_FILE)
     prompt = get_story_start("apocalypse")
     context = get_context("apocalypse")
+
     story_manager = UnconstrainedStoryManager(generator)
     story_manager.start_new_story(prompt, context=context)
 
     print("\n")
+    print(context)
     console_print(str(story_manager.story))
     while (True):
         action = input("> ")
 
         if action != "":
+            action = action.strip()
             action = action[0].lower() + action[1:]
 
             if action[-1] == "." or action[-1] == "?" or action[-1] == "!":
diff --git a/data/app.py b/data/app.py
new file mode 100644
index 0000000..274d64b
--- /dev/null
+++ b/data/app.py
@@ -0,0 +1,13 @@
+import os
+import flask
+import Flask
+app = Flask(__name__)
+
+@app.route("/")
+def hellp_world():
+    target = os.environ.get('TARGET', 'World')
+    return 'Hello {}!\n'.format(target)
+
+
+if __name__ == "__main__":
+    app.run(debug=True, host='0.0.0.0', port=int(os.environ.get('PORT', 8080)))
diff --git a/generator/ctrl/download_model.sh b/generator/ctrl/download_model.sh
index e002f85..3df2c27 100755
--- a/generator/ctrl/download_model.sh
+++ b/generator/ctrl/download_model.sh
@@ -3,6 +3,9 @@ exec -l $SHELL
 gcloud init
 
 
-URL="gs://sf-ctrl/seqlen512_v1.ckpt/"
+URL="gs://aidungeon2model/finetuned_model.tar.gz"
 
 gsutil -m cp -r "$URL" model
+cd model
+tar -xvzf finetuned_model.tar.gz
+cd ..
\ No newline at end of file
diff --git a/generator/tf/models/117M/model.ckpt.index b/generator/tf/models/117M/model.ckpt.index
deleted file mode 100644
index 24729aa..0000000
Binary files a/generator/tf/models/117M/model.ckpt.index and /dev/null differ
diff --git a/generator/tf/models/117M/model.ckpt.meta b/generator/tf/models/117M/model.ckpt.meta
deleted file mode 100644
index 2541738..0000000
Binary files a/generator/tf/models/117M/model.ckpt.meta and /dev/null differ
diff --git a/generator/tf/models/774M/model.ckpt.index b/generator/tf/models/774M/model.ckpt.index
deleted file mode 100644
index 974acbf..0000000
Binary files a/generator/tf/models/774M/model.ckpt.index and /dev/null differ
diff --git a/generator/tf/models/774M/model.ckpt.meta b/generator/tf/models/774M/model.ckpt.meta
deleted file mode 100644
index f5fea4f..0000000
Binary files a/generator/tf/models/774M/model.ckpt.meta and /dev/null differ