Spaces:

vibey
/

article-summariser-for-final-project

Runtime error

vibey commited on Aug 17, 2022

Commit

16fcbec

1 Parent(s): b80af66

Update function.py

Files changed (1) hide show

function.py CHANGED Viewed

@@ -26,7 +26,7 @@ def t5():
 @st.cache(allow_output_mutation=True)
 def ssr():
     ''' Loading ssr model using pipeline api '''
-    summarizer = pipeline("summarization", model="santiviquez/ssr-base-finetuned-samsum-en")
     return summarizer
 def preprocess_plain_text(x):

 @st.cache(allow_output_mutation=True)
 def ssr():
     ''' Loading ssr model using pipeline api '''
+    summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
     return summarizer
 def preprocess_plain_text(x):