FROM my_ocrd_workflow-core-cuda10.0

ARG PIP_INSTALL="pip install --no-cache-dir"
ARG SBB_TEXTLINE_DETECTOR_COMMIT="451b71f"


# Build pip installable stuff
RUN ${PIP_INSTALL} \
    # https://github.com/qurator-spk/sbb_textline_detection/issues/50
    "h5py < 3" \
    https://github.com/qurator-spk/sbb_textline_detector/archive/$SBB_TEXTLINE_DETECTOR_COMMIT.tar.gz


# Copy OCR models
COPY data/textline_detection /var/lib/textline_detection


# Check pip dependencies
RUN pip check


# Default command
CMD ["ocrd-sbb-textline-detector"]