FROM gcr.io/kubeflow-images-public/tensorflow-1.6.0-notebook-cpu
RUN pip install ktext
RUN pip install annoy
RUN pip install --upgrade google-cloud
RUN pip install sklearn h5py
RUN pip install nltk

COPY train.py /workdir/train.py
COPY seq2seq_utils.py /workdir/seq2seq_utils.py