Update README.md
Browse files
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("
|
16 |
-
model = AutoModelForSequenceClassification.from_pretrained("
|
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
|