Commit ·
486e2a6
1
Parent(s): 1501279
Update README.md
Browse files
README.md
CHANGED
|
@@ -27,7 +27,7 @@ The original repository for the paper can be found [here](https://github.com/Bab
|
|
| 27 |
|
| 28 |
## Pipeline usage
|
| 29 |
|
| 30 |
-
```
|
| 31 |
from transformers import pipeline
|
| 32 |
|
| 33 |
triplet_extractor = pipeline('text2text-generation', model='Babelscape/rebel-large', tokenizer='Babelscape/rebel-large')
|
|
@@ -71,7 +71,7 @@ print(extracted_triplets)
|
|
| 71 |
|
| 72 |
## Model and Tokenizer using transformers
|
| 73 |
|
| 74 |
-
```
|
| 75 |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
| 76 |
|
| 77 |
def extract_triplets(text):
|
|
|
|
| 27 |
|
| 28 |
## Pipeline usage
|
| 29 |
|
| 30 |
+
```python
|
| 31 |
from transformers import pipeline
|
| 32 |
|
| 33 |
triplet_extractor = pipeline('text2text-generation', model='Babelscape/rebel-large', tokenizer='Babelscape/rebel-large')
|
|
|
|
| 71 |
|
| 72 |
## Model and Tokenizer using transformers
|
| 73 |
|
| 74 |
+
```python
|
| 75 |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
| 76 |
|
| 77 |
def extract_triplets(text):
|