FROM skin-py-cpu-base:v1.0.2

WORKDIR /app

ENV PYTHONUNBUFFERED=1
ENV PIP_NO_CACHE_DIR=1

COPY wanzheng.py .
COPY api_server.py .
COPY config ./config
COPY best.pt .
COPY shape_predictor_68_face_landmarks.dat .

EXPOSE 8071

CMD ["python", "api_server.py"]

# 服务器执行 nvidia-smi 查看CUDA版本 比如 CUDA 12.1 就是/whl/cu121
# RUN pip install \
#     torch==2.4.1 \
#     torchvision==0.19.1 \
#     --index-url https://download.pytorch.org/whl/cu121