Update function.py

function.py

@@ -22,12 +22,13 @@ def bart():
 def t5():
     ''' Loading t5 model using pipeline api '''
     summarizer = pipeline("summarization", model="t5-base", tokenizer="t5-base")
-    return summarizer    
+    return summarize    
 
-# def pegasus():
-#     ''' Loading pegasus model using pipeline api '''
-#     summarizer = pipeline('summarization',model='google/pegasus-xsum')
-#     return summarizer
+@st.cache(allow_output_mutation=True)
+def ssr():
+    ''' Loading ssr model using pipeline api '''
+    summarizer = pipeline('summarization',model='microsoft/ssr-base')
+    return summarizer
 
 def preprocess_plain_text(x):