NLP-LTU
/

bertweet-large-sexism-detector

Text Classification

Model card Files Files and versions Community

sana-ngu commited on Feb 18, 2023

Commit

72788ff

·

1 Parent(s): 49de308

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -30,7 +30,7 @@ For more details  about the datasets and eval results, see (we will updated the
 from transformers import AutoModelForSequenceClassification, AutoTokenizer,pipeline
 import torch
 model = AutoModelForSequenceClassification.from_pretrained('NLP-LTU/bertweet-large-sexism-detector')
-tokenizer = AutoTokenizer.from_pretrained('vinai/bertweet-large')
 classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)
 prediction=classifier("Every woman wants to be a model. It's codeword for 'I get everything for free and people want me' ")
 label_pred = 'not sexist' if prediction == 0 else 'sexist'

 from transformers import AutoModelForSequenceClassification, AutoTokenizer,pipeline
 import torch
 model = AutoModelForSequenceClassification.from_pretrained('NLP-LTU/bertweet-large-sexism-detector')
+tokenizer = AutoTokenizer.from_pretrained('NLP-LTU/bertweet-large-sexism-detector')
 classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)
 prediction=classifier("Every woman wants to be a model. It's codeword for 'I get everything for free and people want me' ")
 label_pred = 'not sexist' if prediction == 0 else 'sexist'