Update README.md
Browse files
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
|
| 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 |
-
|
|
|
|
| 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 |
```
|