Copycats commited on
Commit
ffd1024
1 Parent(s): 03d49e4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -5,8 +5,8 @@ import torch
5
  from transformers import AutoTokenizer, AutoModelForSequenceClassification, TextClassificationPipeline
6
 
7
  # load model
8
- tokenizer = AutoTokenizer.from_pretrained("jaehyeong/koelectra-base-v3-finetuned-generalized-sentiment-analysis")
9
- model = AutoModelForSequenceClassification.from_pretrained("jaehyeong/koelectra-base-v3-finetuned-generalized-sentiment-analysis")
10
  sentiment_classifier = TextClassificationPipeline(tokenizer=tokenizer, model=model)
11
 
12
  # target reviews
 
5
  from transformers import AutoTokenizer, AutoModelForSequenceClassification, TextClassificationPipeline
6
 
7
  # load model
8
+ tokenizer = AutoTokenizer.from_pretrained("jaehyeong/koelectra-base-v3-generalized-sentiment-analysis")
9
+ model = AutoModelForSequenceClassification.from_pretrained("jaehyeong/koelectra-base-v3-generalized-sentiment-analysis")
10
  sentiment_classifier = TextClassificationPipeline(tokenizer=tokenizer, model=model)
11
 
12
  # target reviews