#!/bin/bash

cd model
# Cython is needed to compile fastBPE
pip install Cython

# Patch the TensorFlow estimator package
export FILE="/usr/local/lib/python2.7/dist-packages/tensorflow_estimator/python/estimator/keras.py"
patch -b "$FILE" estimator.patch

# Install fastBPE
git clone https://github.com/glample/fastBPE.git
cd fastBPE
python setup.py install
cd ../..

# Download the 512-length model if specified, 256-length otherwise
#if [ "$1" = "512" ]
#then
#    URL="gs://sf-ctrl/seqlen512_v1.ckpt/"
#else
#    URL="gs://sf-ctrl/seqlen256_v1.ckpt/"
#fi

# Copy model
#gsutil -m cp -r "$URL" .