russian_art_2024

1
FROM python:3.10-slim
2

3
ARG PIP_DISABLE_PIP_VERSION_CHECK=1
4
ARG PIP_NO_CACHE_DIR=1
5

6
WORKDIR /app
7

8
COPY ./requirements.txt ./requirements.txt
9
RUN pip install -r requirements.txt
10
RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
11

12
COPY ./api/main.py ./main.py
13
COPY ./api/config.py ./config.py
14
COPY ./src/initial_model_utils.py ./initial_model_utils.py
15
COPY ./data/weights/resnet50_tl_68.pt ./weights/resnet50_tl_68.pt
16

17
EXPOSE 8000