brjoey commited on
Commit
5f69ba6
·
verified ·
1 Parent(s): e94da57

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -11,7 +11,7 @@ model_name = "brjoey/climatebert_econbert"
11
  tokenizer = AutoTokenizer.from_pretrained(model_name)
12
  model = AutoModel.from_pretrained(model_name)
13
 
14
- # For sequence classification tasks:
15
  from transformers import AutoModelForSequenceClassification
16
  model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=3)
17
  ```
 
11
  tokenizer = AutoTokenizer.from_pretrained(model_name)
12
  model = AutoModel.from_pretrained(model_name)
13
 
14
+ # or load the model directly with
15
  from transformers import AutoModelForSequenceClassification
16
  model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=3)
17
  ```