Update README.md
Browse files
README.md
CHANGED
|
@@ -21,8 +21,8 @@ Using the model is straightforward with the Hugging Face Transformers library:
|
|
| 21 |
```python
|
| 22 |
from transformers import T5ForConditionalGeneration, T5Tokenizer
|
| 23 |
|
| 24 |
-
model = T5ForConditionalGeneration.from_pretrained("
|
| 25 |
-
tokenizer = T5Tokenizer.from_pretrained("
|
| 26 |
|
| 27 |
def summarize(text):
|
| 28 |
input_text = "summarize: " + text
|
|
|
|
| 21 |
```python
|
| 22 |
from transformers import T5ForConditionalGeneration, T5Tokenizer
|
| 23 |
|
| 24 |
+
model = T5ForConditionalGeneration.from_pretrained("GeorgiaTech/t5-small-finetuned")
|
| 25 |
+
tokenizer = T5Tokenizer.from_pretrained("GeorgiaTech/t5-small-finetuned")
|
| 26 |
|
| 27 |
def summarize(text):
|
| 28 |
input_text = "summarize: " + text
|