vibey commited on
Commit
56bc6ae
1 Parent(s): cb8d7fe

Update function.py

Browse files
Files changed (1) hide show
  1. function.py +1 -1
function.py CHANGED
@@ -18,7 +18,7 @@ def bart():
18
  ''' Loading bart model using pipeline api '''
19
  summarizer = pipeline('summarization',model='facebook/bart-large-cnn')
20
  return summarizer
21
-
22
  def t5():
23
  ''' Loading t5 model using pipeline api '''
24
  summarizer = pipeline("summarization", model="t5-base", tokenizer="t5-base")
 
18
  ''' Loading bart model using pipeline api '''
19
  summarizer = pipeline('summarization',model='facebook/bart-large-cnn')
20
  return summarizer
21
+ @st.cache(allow_output_mutation=True)
22
  def t5():
23
  ''' Loading t5 model using pipeline api '''
24
  summarizer = pipeline("summarization", model="t5-base", tokenizer="t5-base")