thoriqfy's picture
Update README.md
0f16d19
|
raw
history blame
645 Bytes
metadata
language:
  - id
tags:
  - indobert
  - indobenchmark
  - indonlu

How to import:

from transformers import BertForSequenceClassification, BertTokenizer, BertConfig

tokenizer = BertTokenizer.from_pretrained("thoriqfy/indobert-emotion-classification")
config = BertConfig.from_pretrained("thoriqfy/indobert-emotion-classification")
model = BertForSequenceClassification.from_pretrained("thoriqfy/indobert-emotion-classification", config=config)

How to use:

from transformers import pipeline

nlp = pipeline("text-classification", model="thoriqfy/indobert-emotion-classification")

results = nlp("Your input text here")