Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
2
3tokenizer = AutoTokenizer.from_pretrained("data354/camembert-fr-covid-tweet-sentiment-classification")
4model = AutoModelForSequenceClassification.from_pretrained("data354/camembert-fr-covid-tweet-sentiment-classification")
5
6nlp_topic_classif = transformers.pipeline('topics-classification', model = model, tokenizer = tokenizer)
7nlp_topic_classif("tchai on est morts. on va se faire vacciner et ils vont contrôler comme les marionnettes avec des fils. d'après les '' ont dit ''...")
8
9# Output: [{'label': 'opinions', 'score': 0.831]