dream

1
FROM tensorflow/tensorflow:1.14.0-py3
2

3
EXPOSE 8014:8014
4

5
WORKDIR /src
6
RUN mkdir /data /data/models
7

8
RUN  apt-get update \
9
  && apt-get install -y wget \
10
  && rm -rf /var/lib/apt/lists/*
11

12
COPY ./src/requirements.txt requirements.txt
13
RUN pip install --upgrade pip && \
14
    pip install -r requirements.txt
15

16
# COPY ./tfhub_model/ /root/tfhub_cache/
17

18
ENV TFHUB_CACHE_DIR='/root/tfhub_cache'
19
ENV USE_MODEL_PATH='https://tfhub.dev/google/universal-sentence-encoder/1'
20
ENV INTENT_DATA_PATH='/data/intent_data_h3.json'
21
ENV INTENT_PHRASES_PATH='/data/intent_phrases.json'
22
ENV INTENT_MODEL_PATH='/data/models/linear_classifier_h3.h5'
23

24
RUN wget http://lnsigo.mipt.ru/export/alexaprize_data/linear_classifier_h3.h5 -q -P /data/models
25

26
RUN python -c "import tensorflow_hub as hub; model=hub.Module('$USE_MODEL_PATH')"
27

28
COPY ./src/ ./
29
COPY ./data/ /data
30

31
CMD gunicorn --workers=1 --name=catcher --bind 0.0.0.0:8014 --timeout=500 server:app
32