Spaces:

vibey
/

article-summariser-for-final-project

Runtime error

vibey commited on Aug 18, 2022

Commit

40854b9

1 Parent(s): 0bc080c

Update function.py

Files changed (1) hide show

function.py CHANGED Viewed

@@ -26,7 +26,7 @@ def t5():
 def bart_t5():
     ''' Loading bart_t5 model using pipeline api '''
-    summarizer = pipeline("summarization", model="microsoft/ssr-base")
     return summarizer
 # def pegasus():

 def bart_t5():
     ''' Loading bart_t5 model using pipeline api '''
+    summarizer = pipeline("summarization", model="santiviquez/ssr-base-finetuned-samsum-en")
     return summarizer
 # def pegasus():