plippe/faiss-web-service
A web service build on top of Facebook's Faiss
10K+
plippe/faiss-web-service repository overview
Faiss Web Service
Getting started
The fastest way to get started is to use the docker hub image with the following command:
FAISS_VERSION=...
docker run --rm --detach --publish 5000:5000 plippe/faiss-web-service:${FAISS_VERSION}-cpu
Once the container is running, you should be able to ping the service:
# Healthcheck
curl 'localhost:5000/ping'
# Faiss search for ids 1, 2, and 3
curl 'localhost:5000/faiss/search' -X POST -d '{"k": 5, "ids": [1, 2, 3]}'
# Faiss search for a vector
curl 'localhost:5000/faiss/search' -X POST -d '{"k": 5, "vectors": [[54.7, 0.3, 0.6, 0.4, 0.1, 0.7, 0.2, 0.0, 0.6, 0.5, 0.3, 0.2, 0.1, 0.9, 0.3, 0.6, 0.2, 0.9, 0.5, 0.0, 0.9, 0.1, 0.9, 0.1, 0.5, 0.5, 0.8, 0.8, 0.5, 0.2, 0.6, 0.2, 0.2, 0.7, 0.1, 0.7, 0.8, 0.2, 0.9, 0.0, 0.4, 0.4, 0.9, 0.0, 0.6, 0.4, 0.4, 0.6, 0.6, 0.2, 0.5, 0.0, 0.1, 0.6, 0.0, 0.0, 0.4, 0.7, 0.5, 0.7, 0.2, 0.5, 0.5, 0.7]]}'
Custom config
By default, the faiss web service will download files for the faiss index, and for the ids to vectors mapping. This behavior can be overwritten by writting your own configuration file, and having its path as an environment variable FAISS_WEB_SERVICE_CONFIG.
docker run \
--rm \
--tty \
--interactive \
--publish 5000:5000 \
--volume [PATH_TO_YOUR_CONFIG]:/tmp/your_config.py \
--env FAISS_WEB_SERVICE_CONFIG=/tmp/your_config.py \
plippe/faiss-web-service:${FAISS_VERSION}-cpu
Examples of how to write a config file can be found in the resources folder.
Another solution would be to create a new docker image from plippe/faiss-web-service, that sets the environement variable, and adds your config file.
Production
docker run will run the application with Flask's build in server. Flask's documentation clearly states it is not suitable for production. To run the application with uWSGI you must add production to the run command, i.e:
# Flask's build in server
docker run --rm --detach --publish 5000:5000 plippe/faiss-web-service:${FAISS_VERSION}-cpu
# uWSGI server
docker run --rm --detach --publish 5000:5000 plippe/faiss-web-service:${FAISS_VERSION}-cpu production
Tag summary
Content type
Image
Digest
Size
754.7 MB
Last updated
over 8 years ago
docker pull plippe/faiss-web-service