Dockerfile

FROM --platform=linux/amd64 python:3.9-buster

RUN mkdir /data

VOLUME /data

WORKDIR /app

RUN wget -q -O - https://dl-ssl.google.com/linux/linux_signing_key.pub | apt-key add - \
    && sh -c 'echo "deb [arch=amd64] http://dl.google.com/linux/chrome/deb/ stable main" >> /etc/apt/sources.list.d/google-chrome.list' \
    && apt-get -y update \
    && apt-get install -y google-chrome-stable \
    && apt-get install -yqq unzip

RUN wget -O /tmp/chromedriver.zip http://chromedriver.storage.googleapis.com/`curl -sS chromedriver.storage.googleapis.com/LATEST_RELEASE`/chromedriver_linux64.zip \
    && unzip /tmp/chromedriver.zip chromedriver -d /usr/local/bin/

# set display port to avoid crash
ENV DISPLAY=:99

COPY . .

RUN mkdir classifiers 2>/dev/null \
    && wget -P classifiers https://huggingface.co/tsrdjan/scooby/resolve/main/scooby.h5

RUN pip install -r requirements.txt \
    && pip install -e .

CMD ["bash"]