Commit ·
26d06e2
1
Parent(s): 778abea
Update the correct AutoModel
Browse files
README.md
CHANGED
|
@@ -3,9 +3,9 @@ This network has been fine-tuned for the task described in the paper *Topical Ch
|
|
| 3 |
|
| 4 |
# Load the model
|
| 5 |
```python
|
| 6 |
-
from transformers import
|
| 7 |
tokenizer = AutoTokenizer.from_pretrained('dennlinger/roberta-cls-consec')
|
| 8 |
-
model =
|
| 9 |
```
|
| 10 |
|
| 11 |
# Input Format
|
|
|
|
| 3 |
|
| 4 |
# Load the model
|
| 5 |
```python
|
| 6 |
+
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
| 7 |
tokenizer = AutoTokenizer.from_pretrained('dennlinger/roberta-cls-consec')
|
| 8 |
+
model = AutoModelForSequenceClassification.from_pretrained('dennlinger/roberta-cls-consec')
|
| 9 |
```
|
| 10 |
|
| 11 |
# Input Format
|