abdulmatinomotoso
commited on
Commit
•
f2190ac
1
Parent(s):
1dd8afd
Update app.py
Browse files
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("
|
22 |
-
tokenizer = BartTokenizer.from_pretrained("
|
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):
|