Marwolaeth
/

rubert-tiny-nli-terra-v1

Text Classification

natural-language-inference

Inference Endpoints

Model card Files Files and versions Community

Marwolaeth commited on 9 days ago

Commit

29dff30

•

1 Parent(s): db4f38a

A typo in README

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -54,7 +54,7 @@ How to run the model for NLI:
 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
-model_id = 'Marwolaeth/rubert-tiny-nli-terra-v0'
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForSequenceClassification.from_pretrained(model_id)
 if torch.cuda.is_available():

 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
+model_id = 'Marwolaeth/rubert-tiny-nli-terra-v1'
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForSequenceClassification.from_pretrained(model_id)
 if torch.cuda.is_available():