Views
No views yet
1from transformers import AutoModelForSequenceClassification, AutoTokenizer, TextClassificationPipeline
2
3
4model = AutoModelForSequenceClassification.from_pretrained("tsdocode/phobert-finetune-hatespeech", num_labels=3)
5tokenizer = AutoTokenizer.from_pretrained("tsdocode/phobert-finetune-hatespeech")
6
7
8pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer, return_all_scores=True)
9# outputs a list of dicts like [[{'label': 'NEGATIVE', 'score': 0.0001223755971295759}, {'label': 'POSITIVE', 'score': 0.9998776316642761}]]
10pipe("đồ ngu")
11