Update README.md
Browse files
README.md
CHANGED
@@ -29,6 +29,7 @@ To correctly load the full model, include the `trust_remote_code=True` argument
|
|
29 |
## How to use
|
30 |
|
31 |
```python
|
|
|
32 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
33 |
|
34 |
model = AutoModelForSequenceClassification.from_pretrained("manifesto-project/manifestoberta-xlm-roberta-56policy-topics-context-2023-1-1", trust_remote_code=True)
|
|
|
29 |
## How to use
|
30 |
|
31 |
```python
|
32 |
+
import torch
|
33 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
34 |
|
35 |
model = AutoModelForSequenceClassification.from_pretrained("manifesto-project/manifestoberta-xlm-roberta-56policy-topics-context-2023-1-1", trust_remote_code=True)
|