Sentence Similarity
sentence-transformers
PyTorch
Transformers
bert
text-classification
cross-encoder
text-embeddings-inference
Instructions to use jamescalam/bert-stsb-cross-encoder with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use jamescalam/bert-stsb-cross-encoder with sentence-transformers:
from sentence_transformers import CrossEncoder model = CrossEncoder("jamescalam/bert-stsb-cross-encoder") 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) - Transformers
How to use jamescalam/bert-stsb-cross-encoder with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("jamescalam/bert-stsb-cross-encoder") model = AutoModelForSequenceClassification.from_pretrained("jamescalam/bert-stsb-cross-encoder", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Commit History
-
Update model metadata to set pipeline tag to the new `text-ranking`
Verified signatureHello! ## Pull Request overview * Update metadata to set pipeline tag to the new `text-ranking` ## Changes This is an automated pull request to update the metadata of the model card. We recently introduced the [`text-ranking`](https://huggingface.co/models?pipeline_tag=text-ranking) pipeline tag for models that are used for ranking tasks, and we have a suspicion that this model is one of them. Feel free to respond if you have questions or concerns. - Tom Aarsenf6bbf0a
-
Update README.md
6d99739 -
Create README.md
e0582d4
-
added model files for STSb cross encoder demo
ee928bd -
initial commit
0fae9fc