tartuNLP
/

EstBERT128_Rubric

Text Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Community

ksirts commited on Dec 12, 2022

Commit

ad1aacc

·

1 Parent(s): ede5845

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -45,7 +45,7 @@ from transformers import pipeline
 tokenizer = AutoTokenizer.from_pretrained("tartuNLP/EstBERT128_Rubric")
 model = AutoModelForSequenceClassification.from_pretrained("tartuNLP/EstBERT128_Rubric")
-nlp = pipeline("ner", model=bertner, tokenizer=tokenizer)
 text = "Kaia Kanepi (WTA 57.) langes USA-s Charlestonis toimuval WTA 500 kategooria tenniseturniiril konkurentsist kaheksandikfinaalis, kaotades poolatarile Magda Linette'ile (WTA 64.) 3 : 6, 6 : 4, 2 : 6."
 result = nlp(text)

 tokenizer = AutoTokenizer.from_pretrained("tartuNLP/EstBERT128_Rubric")
 model = AutoModelForSequenceClassification.from_pretrained("tartuNLP/EstBERT128_Rubric")
+nlp = pipeline("text-classification", model=model, tokenizer=tokenizer)
 text = "Kaia Kanepi (WTA 57.) langes USA-s Charlestonis toimuval WTA 500 kategooria tenniseturniiril konkurentsist kaheksandikfinaalis, kaotades poolatarile Magda Linette'ile (WTA 64.) 3 : 6, 6 : 4, 2 : 6."
 result = nlp(text)