Emotions_BERT / README.md
cappuch's picture
Update README.md
3e82bce
metadata
language: en
tags:
  - text-classification
  - tensorflow
license: mit

Usage


from transformers import RobertaTokenizerFast, TFRobertaForSequenceClassification, pipeline

tokenizer = RobertaTokenizerFast.from_pretrained("cappuch/Emotions_BERT")
model = TFRobertaForSequenceClassification.from_pretrained("cappuch/Emotions_BERT")

classific = pipeline('sentiment-analysis', 
                    model='cappuch/Emotions_BERT')

labels = classific("Hello!")
print(labels)

#[{'label': 'neutral', 'score': 0.9964383244514465}]