Gbadamosi_oluwaseyi commited on
Commit
5d0397c
1 Parent(s): c501aac
Files changed (1) hide show
  1. function.py +1 -1
function.py CHANGED
@@ -21,7 +21,7 @@ def bart():
21
 
22
  def t5():
23
  ''' Loading t5 model using pipeline api '''
24
- summarizer = pipeline("summarization", model="t5-base", tokenizer="t5-base", framework="tf")
25
  return summarizer
26
 
27
  # def pegasus():
21
 
22
  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 pegasus():