wbrule / README.md
intoxication's picture
Upload 12 files
3fe47db
|
raw
history blame
5.95 kB

Question Answering Application for Healthcare

This is a streamlit-based NLP application powering a question answering demo on healthcare data. It's easy to change and extend and can be used to try out Haystack's capabilities.

A video presentation of this demo is available on YouTube. To get started with Haystack please visit the README or check out our tutorials.

Usage

The easiest way to run the application is through Docker compose. From this folder, just run:

docker compose up -d

Docker will start three containers:

  • elasticsearch, running an Elasticsearch instance with some data pre-loaded.
  • haystack-api, running a pre-loaded Haystack pipeline behind a RESTful API.
  • ui, running the streamlit application showing the UI and querying Haystack under the hood.

Once all the containers are up and running, you can open the user interface pointing your browser to http://localhost:8501.

Screencast

https://user-images.githubusercontent.com/4181769/231965471-48d581a2-e1aa-4316-b3a4-990d9c86800e.mov

Evaluation Mode

The evaluation mode leverages the feedback REST API endpoint of haystack. The user has the options "Wrong answer", "Wrong answer and wrong passage" and "Wrong answer and wrong passage" to give feedback.

In order to use the UI in evaluation mode, you need an ElasticSearch instance with pre-indexed files and the Haystack REST API. You can set the environment up via docker images. For ElasticSearch, you can check out our documentation and for setting up the REST API this [link](https://github.com/deepset-ai/haystack/blob/main/README. md#7-rest-api).

To enter the evaluation mode, select the checkbox "Evaluation mode" in the sidebar. The UI will load the predefined questions from the file [eval_labels_examples](https://raw.githubusercontent.com/ deepset-ai/haystack/main/ui/ui/eval_labels_example.csv). The file needs to be prefilled with your data. This way, the user will get a random question from the set and can give his feedback with the buttons below the questions. To load a new question, click the button "Get random question".

The file just needs to have two columns separated by semicolon. You can add more columns but the UI will ignore them. Every line represents a questions answer pair. The columns with the questions needs to be named “Question Text” and the answer column “Answer” so that they can be loaded correctly. Currently, the easiest way to create the file is manually by adding question answer pairs.

The feedback can be exported with the API endpoint export-doc-qa-feedback. To learn more about finetuning a model with user feedback, please check out our [docs](https://haystack.deepset.ai/usage/ domain-adaptation#user-feedback).

Query different data

If you want to use this application to query a different corpus, the easiest way is to build the Elasticsearch image, load your own text data and then use the same Compose file to run all the three containers needed. This will require Docker to be properly installed on your machine.

Running your custom build

Once done, modify the elasticsearch section in the docker-compose.yml file, changing this line:

 image: "julianrisch/elasticsearch-healthcare"

to:

 image: "my-docker-acct/elasticsearch-custom"

Finally, run the compose file as usual:

docker-compose up

Development

If you want to change the streamlit application, you need to setup your Python environment first. From a virtual environment, run:

pip install -e .

The app requires the Haystack RESTful API to be ready and accepting connections at http://localhost:8000, you can use Docker compose to start only the required containers:

docker-compose up elasticsearch haystack-api

At this point you should be able to make changes and run the streamlit application with:

streamlit run ui/webapp.py

Using GPUs with Docker

Assuming you have nvidia drivers installed on your machine, you can configure docker to use the GPU for the Haystack API container to speed it up. First, configure the nvidia repository as described here: https://nvidia.github.io/nvidia-container-runtime/. For example:

curl -s -L https://nvidia.github.io/nvidia-container-runtime/gpgkey | \
  sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.list | \
  sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list
sudo apt-get update

Then, install nvidia-container-runtime as described here: https://docs.docker.com/config/containers/resource_constraints/#access-an-nvidia-gpu. For example:

sudo apt-get install nvidia-container-runtime

Restart the Docker daemon (or simply the machine). Finally, you can change the docker compose file healthcare/docker-compose.yml so that a docker image prepared for usage with GPUs is used and one GPU is reserved for the Haystack API container:

  haystack-api:
    image: "deepset/haystack:gpu-v1.14.0"
    ports:
      - 8000:8000
    restart: on-failure
    volumes:
      - ./haystack-api:/home/node/app
    environment:
      - DOCUMENTSTORE_PARAMS_HOST=elasticsearch
      - PIPELINE_YAML_PATH=/home/node/app/pipelines_biobert.haystack-pipeline.yml
    depends_on:
      elasticsearch:
        condition: service_healthy
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]