nfhakim's picture
Update README.md
75b027e verified
|
raw
history blame
632 Bytes

This is the first classification of sentiment analysis for (redacted) task

How to import

import torch
from transformers import BertForSequenceClassification, BertTokenizer, BertConfig

tokenizer = BertTokenizer.from_pretrained("nfhakim/sentiment-analysis-c1")
config = BertConfig.from_pretrained("nfhakim/sentiment-analysis-c1")
model = BertForSequenceClassification.from_pretrained("nfhakim/sentiment-analysis-c1", config=config)

How to use

from transformers import pipeline
nlp = pipeline("text-classification", model="nfhakim/sentiment-analysis-c1")

results = nlp("Your input text here")