vibey commited on
Commit
64b9cf1
1 Parent(s): 6569c40

Update function.py

Browse files
Files changed (1) hide show
  1. function.py +2 -2
function.py CHANGED
@@ -24,8 +24,8 @@ def t5():
24
  summarizer = pipeline("summarization", model="t5-base", tokenizer="t5-base")
25
  return summarizer
26
  @st.cache(allow_output_mutation=True)
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
 
24
  summarizer = pipeline("summarization", model="t5-base", tokenizer="t5-base")
25
  return summarizer
26
  @st.cache(allow_output_mutation=True)
27
+ def ssr():
28
+ ''' Loading ssr model using pipeline api '''
29
  summarizer = pipeline("summarization", model="santiviquez/ssr-base-finetuned-samsum-en")
30
  return summarizer
31