abdulmatinomotoso commited on
Commit
f2190ac
1 Parent(s): 1dd8afd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,8 +18,8 @@ def read_in_text(url):
18
  #initailizing the model pipeline
19
  from transformers import BartTokenizer, BartForConditionalGeneration
20
 
21
- model = BartForConditionalGeneration.from_pretrained("facebook/bart-large-cnn")
22
- tokenizer = BartTokenizer.from_pretrained("facebook/bart-large-cnn")
23
 
24
  #Defining a function to get the summary of the article
25
  def final_summary(file):
 
18
  #initailizing the model pipeline
19
  from transformers import BartTokenizer, BartForConditionalGeneration
20
 
21
+ model = BartForConditionalGeneration.from_pretrained("sshleifer/distilbart-cnn-12-6")
22
+ tokenizer = BartTokenizer.from_pretrained("sshleifer/distilbart-cnn-12-6")
23
 
24
  #Defining a function to get the summary of the article
25
  def final_summary(file):