radu-corcel/reranker

fork of Cross-Encoder for MS Marco

Original weights: https://huggingface.co/cross-encoder/ms-marco-MiniLM-L-12-v2

Added a custom handler to do reranking: https://www.sbert.net/docs/cross_encoder/usage/usage.html

Example json usage:

{
  "inputs": {
    "query": "I like Apples",
    "documents": [
       "I add red ones to smoothies",
       "Red ones are larger than green ones",
       "Asia is a continent"
    ]
  },
  "parameters": {} // not used yet
}