Update README.md
Browse files
README.md
CHANGED
|
@@ -73,7 +73,7 @@ Replace with your best numbers:
|
|
| 73 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 74 |
import torch
|
| 75 |
|
| 76 |
-
model_name = "
|
| 77 |
tok = AutoTokenizer.from_pretrained(model_name)
|
| 78 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
| 79 |
|
|
|
|
| 73 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 74 |
import torch
|
| 75 |
|
| 76 |
+
model_name = "sdeakin/fine_tuned_bert_emotions_large"
|
| 77 |
tok = AutoTokenizer.from_pretrained(model_name)
|
| 78 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
| 79 |
|