Update README.md
Browse files
README.md
CHANGED
@@ -4,7 +4,7 @@ The model is trained on TV and Movie datasets and takes simplified Chinese as in
|
|
4 |
We trained the model from the "hfl/chinese-bert-wwm-ext" checkpoint.
|
5 |
|
6 |
#### Sample Usage
|
7 |
-
from transformers import
|
8 |
|
9 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
10 |
checkpoint = "Herais/pred_genre"
|
|
|
4 |
We trained the model from the "hfl/chinese-bert-wwm-ext" checkpoint.
|
5 |
|
6 |
#### Sample Usage
|
7 |
+
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
8 |
|
9 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
10 |
checkpoint = "Herais/pred_genre"
|