Views
No views yet
bias-model-stereoset1 → Biased sentence0 → Not biased sentencebert-base-uncased1from transformers import BertTokenizer, BertForSequenceClassification
2
3tokenizer = BertTokenizer.from_pretrained("AymanKhan/bias-model-stereoset")
4model = BertForSequenceClassification.from_pretrained("AymanKhan/bias-model-stereoset")
5
6inputs = tokenizer("The people are fat and unathletic.", return_tensors="pt")
7outputs = model(**inputs)
8pred = outputs.logits.argmax(dim=1).item()
9
10print("🔴 Biased" if pred == 1 else "🟢 Not Biased")