Commit
·
6382be9
1
Parent(s):
a7aef0d
Update model card
Browse files
README.md
CHANGED
@@ -24,8 +24,8 @@ English gathered from social media platforms: Twitter and Gab. It achieved a **M
|
|
24 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
|
25 |
|
26 |
# Load model and tokenizer
|
27 |
-
tokenizer =
|
28 |
-
model =
|
29 |
|
30 |
# Create the pipeline for classification
|
31 |
sexism_classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)
|
|
|
24 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
|
25 |
|
26 |
# Load model and tokenizer
|
27 |
+
tokenizer = AutoTokenizer.from_pretrained('tum-nlp/bertweet-sexism')
|
28 |
+
model = AutoModelForSequenceClassification.from_pretrained('tum-nlp/bertweet-sexism')
|
29 |
|
30 |
# Create the pipeline for classification
|
31 |
sexism_classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)
|