Forturne commited on
Commit
9c0f7fd
·
1 Parent(s): 65527bc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -3,8 +3,8 @@ You can use this model with Transformers pipeline for sentiment analysis.
3
  from transformers import BertTokenizer, BertForSequenceClassification
4
  from transformers import pipeline
5
 
6
- finbert = BertForSequenceClassification.from_pretrained('yiyanghkust/finbert-tone',num_labels=3)
7
- tokenizer = BertTokenizer.from_pretrained('yiyanghkust/finbert-tone')
8
 
9
  nlp = pipeline("sentiment-analysis", model=finbert, tokenizer=tokenizer)
10
 
 
3
  from transformers import BertTokenizer, BertForSequenceClassification
4
  from transformers import pipeline
5
 
6
+ finbert = BertForSequenceClassification.from_pretrained('Forturne/Finbert_PB',num_labels=3)
7
+ tokenizer = BertTokenizer.from_pretrained('Forturne/Finbert_PB')
8
 
9
  nlp = pipeline("sentiment-analysis", model=finbert, tokenizer=tokenizer)
10