File size: 794 Bytes
0c5ad16 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
FROM python:3.11 as base-img
ARG DEBIAN_FRONTEND=noninteractive
RUN apt-get update && apt-get install -y python3 ffmpeg libxext6 libsm6 libgtk2.0-dev pkg-config libgtk2.0-dev libavcodec-dev python3-opencv libopencv-dev
RUN apt-get install -y python3-pip
RUN pip3 install torch torchvision
FROM base-img as app
RUN useradd -m -u 1000 user
# Switch to the "user" user
USER user
# Set home to the user's home directory
ENV HOME=/home/user \
PATH=/home/user/.local/bin:$PATH
WORKDIR $HOME/app
# Copy the current directory contents into the container at $HOME/app setting the owner to the user
COPY --chown=user requirements.txt requirements.txt
COPY --chown=user data data
COPY --chown=user detr detr
RUN pip3 install -r requirements.txt
ENTRYPOINT [ "python","detr/main_gradio.py" ] |