jaimin commited on
Commit
e48c2f7
Β·
1 Parent(s): 9c30e19

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -32,8 +32,8 @@ def clean_text(url):
32
  #initailizing the model pipeline
33
  from transformers import BartTokenizer, BartForConditionalGeneration
34
 
35
- model = BartForConditionalGeneration.from_pretrained("sshleifer/distilbart-cnn-12-6")
36
- tokenizer = BartTokenizer.from_pretrained("sshleifer/distilbart-cnn-12-6")
37
  nlp = spacy.load("en_core_web_sm")
38
 
39
  #Defining a function to get the summary of the article
 
32
  #initailizing the model pipeline
33
  from transformers import BartTokenizer, BartForConditionalGeneration
34
 
35
+ model = BartForConditionalGeneration.from_pretrained("jaimin/distilbart-summarizer")
36
+ tokenizer = BartTokenizer.from_pretrained("jaimin/distilbart-summarizer")
37
  nlp = spacy.load("en_core_web_sm")
38
 
39
  #Defining a function to get the summary of the article