Spaces:

vibey
/

article-summariser-for-final-project

Runtime error

vibey commited on Aug 17, 2022

Commit

36a8222

1 Parent(s): cffd6ee

Update function.py

Files changed (1) hide show

function.py CHANGED Viewed

@@ -27,7 +27,7 @@ def t5():
 @st.cache(allow_output_mutation=True)
 def ssr():
     ''' Loading ssr model using pipeline api '''
-    summarizer = pipeline("summarization", model="microsoft/ssr-base", tokenizer="microsoft/ssr-base")
     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="ssr-base", tokenizer="ssr-base")
     return summarizer
 def preprocess_plain_text(x):