Update README.md
Browse files
README.md
CHANGED
|
@@ -17,9 +17,9 @@ propose a unified model in the biomedical domain and use instructions to achieve
|
|
| 17 |
How to Use
|
| 18 |
=============
|
| 19 |
|
| 20 |
-
You can very easily load the models with Transformers, instead of downloading them manually. BART-base model is the backbone of our model.
|
| 21 |
|
| 22 |
-
```
|
| 23 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
| 24 |
|
| 25 |
tokenizer = AutoTokenizer.from_pretrained("cogint/in-boxbart")
|
|
|
|
| 17 |
How to Use
|
| 18 |
=============
|
| 19 |
|
| 20 |
+
You can very easily load the models with Transformers, instead of downloading them manually. The BART-base model is the backbone of our model. Here is how to use the model in PyTorch:
|
| 21 |
|
| 22 |
+
```python
|
| 23 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
| 24 |
|
| 25 |
tokenizer = AutoTokenizer.from_pretrained("cogint/in-boxbart")
|