How to use ncbi/MedCPT-Cross-Encoder with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="ncbi/MedCPT-Cross-Encoder")
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("ncbi/MedCPT-Cross-Encoder") model = AutoModelForSequenceClassification.from_pretrained("ncbi/MedCPT-Cross-Encoder")
2bd0534
1
2
3
4
5
6
7
8
{ "[CLS]": 2, "[MASK]": 4, "[PAD]": 0, "[SEP]": 3, "[UNK]": 1 }
