Devika Nair M
commited on
Update Briefly.py
Browse files- Briefly.py +1 -1
Briefly.py
CHANGED
@@ -9,7 +9,7 @@ from annotated_text import annotated_text # to display keywords
|
|
9 |
# Load sshleifer/distilbart-cnn-12-6 model
|
10 |
@st.cache(allow_output_mutation=True)
|
11 |
def load_model():
|
12 |
-
model = pipeline
|
13 |
return model
|
14 |
|
15 |
data = gnewsclient.NewsClient(max_results=0)
|
|
|
9 |
# Load sshleifer/distilbart-cnn-12-6 model
|
10 |
@st.cache(allow_output_mutation=True)
|
11 |
def load_model():
|
12 |
+
model = pipeline("summarization")
|
13 |
return model
|
14 |
|
15 |
data = gnewsclient.NewsClient(max_results=0)
|