Update function.py

function.py

@@ -22,12 +22,17 @@ def bart():
 def t5():
     ''' Loading t5 model using pipeline api '''
     summarizer = pipeline("summarization", model="t5-base", tokenizer="t5-base")
-    return summarizer    
-@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 bart_t5():
+    ''' Loading bart_t5 model using pipeline api '''
+    summarizer = pipeline("summarization", model="tuner007/pegasus_summarizer")
+    return summarizer     
+
+# def pegasus():
+#     ''' Loading pegasus model using pipeline api '''
+#     summarizer = pipeline('summarization',model='google/pegasus-xsum')
+#     return summarizer
 
 def preprocess_plain_text(x):