Views
No views yet
1from transformers import AutoModelForSequenceClassification, AutoTokenizer, TextClassificationPipeline
2
3model_path = "ZiruiXiong/bert-base-finetuned-toxic-comment-classification"
4tokenizer = AutoTokenizer.from_pretrained(model_path)
5model = AutoModelForSequenceClassification.from_pretrained(model_path)
6
7pipeline = TextClassificationPipeline(model=model, tokenizer=tokenizer)
8print(pipeline('You are also a liar since I hardly spoke with you.'))