richardorama
commited on
Commit
•
0e5f8d9
1
Parent(s):
2c0ab44
Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,8 @@ import streamlit as st
|
|
5 |
from transformers import pipeline
|
6 |
|
7 |
# Load the summarization model
|
8 |
-
summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6") # smaller version of the model
|
|
|
9 |
|
10 |
# Default article text
|
11 |
DEFAULT_ARTICLE = """ New York (CNN)When Liana Barrientos was 23 years old, she got married in Westchester County, New York.
|
|
|
5 |
from transformers import pipeline
|
6 |
|
7 |
# Load the summarization model
|
8 |
+
#summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6") # smaller version of the model
|
9 |
+
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
|
10 |
|
11 |
# Default article text
|
12 |
DEFAULT_ARTICLE = """ New York (CNN)When Liana Barrientos was 23 years old, she got married in Westchester County, New York.
|