vibey commited on
Commit
6569c40
1 Parent(s): 40854b9

Update function.py

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