cointegrated/nli-rus-translated-v2021
Viewer • Updated • 1.9M • 172 • 1
How to use cointegrated/rubert-base-cased-nli-twoway with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("zero-shot-classification", model="cointegrated/rubert-base-cased-nli-twoway") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("cointegrated/rubert-base-cased-nli-twoway")
model = AutoModelForSequenceClassification.from_pretrained("cointegrated/rubert-base-cased-nli-twoway")This is the DeepPavlov/rubert-base-cased fine-tuned to predict the logical relationship between two short texts: entailment or not entailment.
For more details, see the card for a similar model: https://huggingface.co/cointegrated/rubert-base-cased-nli-threeway