add reranker (#679)

This commit is contained in:
luanshaotong
2024-01-02 18:10:15 +08:00
committed by GitHub
parent 2e75851b02
commit d5b24eca57
4 changed files with 171 additions and 0 deletions

View File

@@ -0,0 +1,10 @@
FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
# please download the model from https://huggingface.co/BAAI/bge-reranker-base and put it in the same directory as Dockerfile
COPY ./bge-reranker-base ./bge-reranker-base
COPY app.py Dockerfile requirement.txt .
RUN python3 -m pip install -r requirement.txt
ENTRYPOINT python3 app.py