Gbadamosi_oluwaseyi commited on
Commit
ffb4029
1 Parent(s): b495582
Files changed (1) hide show
  1. function.py +1 -1
function.py CHANGED
@@ -16,7 +16,7 @@ time_str = time.strftime("%Y%m%d-%H%M%S")
16
  @st.cache(allow_output_mutation=True)
17
  def bart():
18
  ''' Loading bart model using pipeline api '''
19
- summarizer = pipeline('summarization',model='amazon/bort')
20
  return summarizer
21
 
22
  def t5():
 
16
  @st.cache(allow_output_mutation=True)
17
  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():