Elron
/

bleurt-large-512

Text Classification

text-embeddings-inference

Model card Files Files and versions

Elron commited on Oct 4, 2021

Commit

aa9bd80

·

1 Parent(s): 1c191c4

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -11,8 +11,8 @@ The code for model conversion was originated from [this notebook](https://colab.
 from transformers import BertForSequenceClassification, AutoTokenizer
 import torch
-tokenizer = AutoTokenizer.from_pretrained("elron/bleurt-large-512")
-model = AutoModelForSequenceClassification.from_pretrained("elron/bleurt-large-512")
 model.eval()
 references = ["hello world", "hello world"]

 from transformers import BertForSequenceClassification, AutoTokenizer
 import torch
+tokenizer = AutoTokenizer.from_pretrained("Elron/bleurt-large-512")
+model = AutoModelForSequenceClassification.from_pretrained("Elron/bleurt-large-512")
 model.eval()
 references = ["hello world", "hello world"]