vibey commited on
Commit
b22161f
·
1 Parent(s): b990b87

Update function.py

Browse files
Files changed (1) hide show
  1. function.py +1 -6
function.py CHANGED
@@ -22,12 +22,7 @@ def bart():
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 bart_t5():
28
- ''' Loading bart_t5 model using pipeline api '''
29
- summarizer = pipeline("summarization", model="tuner007/pegasus_summarizer")
30
- return summarizer
31
 
32
  # def pegasus():
33
  # ''' Loading pegasus model using pipeline api '''
 
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():
28
  # ''' Loading pegasus model using pipeline api '''