Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2
3model_id = "aryaman1222/safeornot-safety-model"
4
5tokenizer = AutoTokenizer.from_pretrained(model_id)
6model = AutoModelForSequenceClassification.from_pretrained(model_id)