Copycats commited on
Commit
370f325
·
verified ·
1 Parent(s): 536e2d3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -12,8 +12,8 @@ import torch
12
  from transformers import AutoTokenizer, AutoModelForSequenceClassification, TextClassificationPipeline
13
 
14
  # load model
15
- tokenizer = AutoTokenizer.from_pretrained("jaehyeong/koelectra-base-v3-generalized-sentiment-analysis")
16
- model = AutoModelForSequenceClassification.from_pretrained("jaehyeong/koelectra-base-v3-generalized-sentiment-analysis")
17
  sentiment_classifier = TextClassificationPipeline(tokenizer=tokenizer, model=model)
18
 
19
  # target reviews
 
12
  from transformers import AutoTokenizer, AutoModelForSequenceClassification, TextClassificationPipeline
13
 
14
  # load model
15
+ tokenizer = AutoTokenizer.from_pretrained("Copycats/koelectra-base-v3-generalized-sentiment-analysis")
16
+ model = AutoModelForSequenceClassification.from_pretrained("Copycats/koelectra-base-v3-generalized-sentiment-analysis")
17
  sentiment_classifier = TextClassificationPipeline(tokenizer=tokenizer, model=model)
18
 
19
  # target reviews