Update function.py

function.py

@@ -23,6 +23,11 @@ def t5():
     ''' Loading t5 model using pipeline api '''
     summarizer = pipeline("summarization", model="t5-base", tokenizer="t5-base")
     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 '''