Update function.py

function.py

@@ -23,11 +23,10 @@ def t5():
     ''' Loading t5 model using pipeline api '''
     summarizer = pipeline("summarization", model="t5-base", tokenizer="t5-base")
     return summarize    
-
 @st.cache(allow_output_mutation=True)
 def ssr():
     ''' Loading ssr model using pipeline api '''
-    summarizer = pipeline("summarization", model="ssr-base", tokenizer="ssr-base")
+    summarizer = pipeline("summarization", model="microsoft/ssr-base")
     return summarizer
 
 def preprocess_plain_text(x):