FROM python:3.9-slim-bullseye

RUN pip install --upgrade pip
RUN pip install facenet-pytorch

# Create working directory
RUN mkdir -p /usr/src/app
WORKDIR /usr/src/app

# Copy weights
ENV TORCH_HOME=/weights
COPY /weights /weights

# Copy source code
COPY face.py /usr/src/app

ENTRYPOINT ["python3", "/usr/src/app/face.py", "--input", "/in", "--output", "/out"]