tburst commited on
Commit
2696ffd
1 Parent(s): 9a8ca27

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
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)