david-inf commited on
Commit
a9958d4
·
verified ·
1 Parent(s): 3caaf51

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -10,15 +10,15 @@ library_name: transformers
10
  ---
11
 
12
  ### Model summary
13
- `bert-sci-am` is a BERT-family model trained for scientific literature argument mining. At low-level it performs sentence classification.
14
 
15
  ### How to use
16
 
17
  ```python
18
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
19
  checkpoint = "david-inf/bert-sci-am"
20
- device = "cuda"
21
 
22
- model =
23
- tokenizer =
 
24
  ```
 
10
  ---
11
 
12
  ### Model summary
13
+ `bert-sci-am` is a BERT-family model trained for scientific literature argument mining. At low-level it performs sequence classification.
14
 
15
  ### How to use
16
 
17
  ```python
18
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
19
  checkpoint = "david-inf/bert-sci-am"
 
20
 
21
+ model = AutoModelForSequenceClassification.from_pretrained(
22
+ "david-inf/bert-sci-am", num_labels=2)
23
+ tokenizer = AutoTokenizer.from_pretrained("david-inf/bert-sci-am")
24
  ```