mlsdp/chapter4_serving_patterns/data_cache_pattern/Dockerfile.proxy at main · wikibook/mlsdp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
FROM python:3.8-slim as builder

ARG SERVER_DIR=resnet50_onnx_runtime
ENV PROJECT_DIR data_cache_pattern

WORKDIR /${PROJECT_DIR}
ADD ./${SERVER_DIR}/requirements.txt /${PROJECT_DIR}/

COPY ./${SERVER_DIR}/extract_resnet50_onnx.py /${PROJECT_DIR}/extract_resnet50_onnx.py
COPY ./src/ml/transformers.py /${PROJECT_DIR}/src/ml/transformers.py
COPY ./data /${PROJECT_DIR}/data

RUN apt-get -y update && \
    apt-get -y install apt-utils gcc && \
    apt-get clean && \
    rm -rf /var/lib/apt/lists/* && \
    pip install --no-cache-dir -r requirements.txt && \
    touch __init__.py && \
    touch src/__init__.py && \
    touch src/ml/__init__.py && \
    python -m extract_resnet50_onnx --prep


FROM python:3.8-slim

ENV PROJECT_DIR data_cache_pattern
ENV MODEL_BASE_PATH=/${PROJECT_DIR}/models

WORKDIR /${PROJECT_DIR}
ADD ./requirements.txt /${PROJECT_DIR}/
RUN apt-get -y update && \
    apt-get -y install apt-utils gcc && \
    apt-get clean && \
    rm -rf /var/lib/apt/lists/* && \
    pip install --no-cache-dir -r requirements.txt

COPY ./src/ /${PROJECT_DIR}/src/
COPY --from=builder /${MODEL_BASE_PATH}/preprocess_transformer.pkl ${MODEL_BASE_PATH}/preprocess_transformer.pkl
COPY --from=builder /${MODEL_BASE_PATH}/softmax_transformer.pkl ${MODEL_BASE_PATH}/softmax_transformer.pkl
COPY ./data /${PROJECT_DIR}/data

ENV PREPROCESS_TRANSFORMER_PATH ${MODEL_BASE_PATH}/preprocess_transformer.pkl
ENV SOFTMAX_TRANSFORMER_PATH ${MODEL_BASE_PATH}/softmax_transformer.pkl
ENV SAMPLE_IMAGE_PATH /${PROJECT_DIR}/data/cat.jpg
ENV LABEL_PATH /${PROJECT_DIR}/data/image_net_labels.json

ENV LOG_LEVEL DEBUG
ENV LOG_FORMAT TEXT

COPY ./run.sh /${PROJECT_DIR}/run.sh
RUN chmod +x /${PROJECT_DIR}/run.sh
CMD [ "./run.sh" ]