FROM python:3.10

RUN pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu
RUN pip install --user 'git+https://github.com/facebookresearch/detectron2.git'
RUN apt-get update
RUN apt-get install ffmpeg libsm6 libxext6 -y
RUN pip install opencv-python==4.6.0.66 pymongo==4.2.0

COPY ../model_v1 output/
COPY predict.py .

ENTRYPOINT python predict.py