vibey commited on
Commit
40854b9
1 Parent(s): 0bc080c

Update function.py

Browse files
Files changed (1) hide show
  1. function.py +1 -1
function.py CHANGED
@@ -26,7 +26,7 @@ def t5():
26
 
27
  def bart_t5():
28
  ''' Loading bart_t5 model using pipeline api '''
29
- summarizer = pipeline("summarization", model="microsoft/ssr-base")
30
  return summarizer
31
 
32
  # def pegasus():
 
26
 
27
  def bart_t5():
28
  ''' Loading bart_t5 model using pipeline api '''
29
+ summarizer = pipeline("summarization", model="santiviquez/ssr-base-finetuned-samsum-en")
30
  return summarizer
31
 
32
  # def pegasus():