Update README.md
Browse files
README.md
CHANGED
@@ -38,8 +38,8 @@ provided a simple snippet of how to use this model for the task of paragraph sum
|
|
38 |
```python
|
39 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
40 |
|
41 |
-
tokenizer = AutoTokenizer.from_pretrained("suriya7/
|
42 |
-
model = AutoModelForSeq2SeqLM.from_pretrained("suriya7/
|
43 |
|
44 |
def generate_summary(text):
|
45 |
inputs = tokenizer([text], max_length=1024, return_tensors='pt', truncation=True)
|
|
|
38 |
```python
|
39 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
40 |
|
41 |
+
tokenizer = AutoTokenizer.from_pretrained("suriya7/bart-finetuned-text-summarization")
|
42 |
+
model = AutoModelForSeq2SeqLM.from_pretrained("suriya7/bart-finetuned-text-summarization")
|
43 |
|
44 |
def generate_summary(text):
|
45 |
inputs = tokenizer([text], max_length=1024, return_tensors='pt', truncation=True)
|