mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-12-30 00:30:09 +00:00
9c3d9b4885
* Add utility to cache models and nltk data & modify Dockerfiles to use it * Fix punkt data not being cached
46 lines
1.4 KiB
Docker
46 lines
1.4 KiB
Docker
FROM python:3.7.4-stretch
|
|
|
|
WORKDIR /home/user
|
|
|
|
RUN apt-get update && apt-get install -y curl git pkg-config cmake
|
|
|
|
# Install PDF converter
|
|
RUN wget --no-check-certificate https://dl.xpdfreader.com/xpdf-tools-linux-4.03.tar.gz && \
|
|
tar -xvf xpdf-tools-linux-4.03.tar.gz && cp xpdf-tools-linux-4.03/bin64/pdftotext /usr/local/bin
|
|
|
|
RUN apt-get install libpoppler-cpp-dev pkg-config -y --fix-missing
|
|
|
|
# Install Tesseract
|
|
RUN apt-get install tesseract-ocr libtesseract-dev poppler-utils -y
|
|
|
|
# copy code
|
|
COPY haystack /home/user/haystack
|
|
|
|
# install as a package
|
|
COPY setup.py requirements.txt README.md /home/user/
|
|
RUN pip install --upgrade pip
|
|
RUN pip install -r requirements.txt
|
|
RUN pip install -e .
|
|
RUN python3 -c "from haystack.utils.docker import cache_models;cache_models()"
|
|
|
|
# create folder for /file-upload API endpoint with write permissions, this might be adjusted depending on FILE_UPLOAD_PATH
|
|
RUN mkdir -p /home/user/file-upload
|
|
RUN chmod 777 /home/user/file-upload
|
|
|
|
# copy saved models
|
|
COPY README.md models* /home/user/models/
|
|
|
|
# Copy REST API code
|
|
COPY rest_api /home/user/rest_api
|
|
|
|
# optional : copy sqlite db if needed for testing
|
|
#COPY qa.db /home/user/
|
|
|
|
# optional: copy data directory containing docs for ingestion
|
|
#COPY data /home/user/data
|
|
|
|
EXPOSE 8000
|
|
|
|
# cmd for running the API
|
|
CMD ["gunicorn", "rest_api.application:app", "-b", "0.0.0.0", "-k", "uvicorn.workers.UvicornWorker", "--workers", "1", "--timeout", "180"]
|