Instructions to use bmilon/rerank-based-on-BAAI with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use bmilon/rerank-based-on-BAAI with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("bmilon/rerank-based-on-BAAI") model = AutoModelForSequenceClassification.from_pretrained("bmilon/rerank-based-on-BAAI") - sentence-transformers
How to use bmilon/rerank-based-on-BAAI with sentence-transformers:
from sentence_transformers import CrossEncoder model = CrossEncoder("bmilon/rerank-based-on-BAAI") query = "Which planet is known as the Red Planet?" passages = [ "Venus is often called Earth's twin because of its similar size and proximity.", "Mars, known for its reddish appearance, is often referred to as the Red Planet.", "Jupiter, the largest planet in our solar system, has a prominent red spot.", "Saturn, famous for its rings, is sometimes mistaken for the Red Planet." ] scores = model.predict([(query, passage) for passage in passages]) print(scores) - Notebooks
- Google Colab
- Kaggle
File size: 135 Bytes
194174c | 1 2 3 4 | version https://git-lfs.github.com/spec/v1
oid sha256:3dfa77cef865d2873256b2678c09be55c0e742e75fc9d01b5fc3c2291bd48fd1
size 2239614572
|