suriya7 commited on
Commit
31a8954
·
verified ·
1 Parent(s): 5cc57b1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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/text_summarize")
42
- model = AutoModelForSeq2SeqLM.from_pretrained("suriya7/text_summarize")
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)