Update README.md
Browse files
README.md
CHANGED
|
@@ -24,7 +24,7 @@ print(tokenizer.batch_decode(outputs, skip_special_tokens=True))
|
|
| 24 |
|
| 25 |
```python
|
| 26 |
from transformers import pipeline
|
| 27 |
-
srl = pipeline("cu-kairos/propbank_srl_seq2seq_t5_large")
|
| 28 |
print(srl(["SRL for [put]: That fund was [put] together by Blackstone Group ."]))
|
| 29 |
|
| 30 |
# [{'generated_text': 'ARG-1: That fund | ARG-2: together | ARG-0: by Blackstone Group '}]
|
|
|
|
| 24 |
|
| 25 |
```python
|
| 26 |
from transformers import pipeline
|
| 27 |
+
srl = pipeline("text2text-generation", "cu-kairos/propbank_srl_seq2seq_t5_large")
|
| 28 |
print(srl(["SRL for [put]: That fund was [put] together by Blackstone Group ."]))
|
| 29 |
|
| 30 |
# [{'generated_text': 'ARG-1: That fund | ARG-2: together | ARG-0: by Blackstone Group '}]
|