mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-11-01 10:19:23 +00:00
36 lines
1.5 KiB
YAML
36 lines
1.5 KiB
YAML
version: '3'
|
|
services:
|
|
haystack-api:
|
|
build:
|
|
context: .
|
|
dockerfile: Dockerfile
|
|
image: "deepset/haystack-cpu:latest"
|
|
ports:
|
|
- 8000:8000
|
|
volumes:
|
|
# Optional: mount your own models from disk into the container
|
|
- "./models:/home/user/models"
|
|
environment:
|
|
# See rest_api/config.py for more variables that you can configure here.
|
|
- DB_HOST=elasticsearch
|
|
- USE_GPU=False
|
|
# Load a model from transformers' model hub or a local path into the FARMReader.
|
|
- READER_MODEL_PATH=deepset/roberta-base-squad2
|
|
# - READER_MODEL_PATH=home/user/models/roberta-base-squad2
|
|
# Alternative: If you want to use the TransformersReader (e.g. for loading a local model in transformers format):
|
|
# - READER_TYPE=TransformersReader
|
|
# - READER_MODEL_PATH=/home/user/models/roberta-base-squad2
|
|
# - READER_TOKENIZER=/home/user/models/roberta-base-squad2
|
|
restart: always
|
|
depends_on:
|
|
- elasticsearch
|
|
command: "/bin/bash -c 'sleep 15 && gunicorn rest_api.application:app -b 0.0.0.0 -k uvicorn.workers.UvicornWorker --workers 1 --timeout 180'"
|
|
elasticsearch:
|
|
# This will start an empty elasticsearch instance (so you have to add your documents yourself)
|
|
image: "elasticsearch:7.6.1"
|
|
# If you want a demo image instead that is "ready-to-query" with some indexed Game of Thrones articles:
|
|
# image: "deepset/elasticsearch-game-of-thrones"
|
|
ports:
|
|
- 9200:9200
|
|
environment:
|
|
- discovery.type=single-node |