From d6f3f46b2082ec5257e49abe4912cef516912633 Mon Sep 17 00:00:00 2001
From: Nick <nickwalton00@gmail.com>
Date: Wed, 18 Sep 2019 13:14:37 -0600
Subject: [PATCH] updated script

---
 generator/ctrl/install_ctrl.sh | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/generator/ctrl/install_ctrl.sh b/generator/ctrl/install_ctrl.sh
index 6ecb093..3667751 100755
--- a/generator/ctrl/install_ctrl.sh
+++ b/generator/ctrl/install_ctrl.sh
@@ -4,6 +4,9 @@ cd model
 # Cython is needed to compile fastBPE
 pip install Cython
 
+pip install tensorflow-gpu==1.14
+pip install keras
+
 # Patch the TensorFlow estimator package
 FILE="/usr/local/lib/python3.6/dist-packages/tensorflow_estimator/python/estimator/keras.py"
 patch -b "$FILE" estimator.patch
@@ -14,8 +17,6 @@ cd fastBPE
 python setup.py install
 cd ..
 
-pip install tensorflow-gpu==1.14
-
 curl https://sdk.cloud.google.com | bash
 exec -l $SHELL
 gcloud init