vibey commited on
Commit
faa6429
·
1 Parent(s): dca0014

Update function.py

Browse files
Files changed (1) hide show
  1. function.py +1 -1
function.py CHANGED
@@ -22,7 +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 summarize
26
  @st.cache(allow_output_mutation=True)
27
  def ssr():
28
  ''' Loading ssr 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
  @st.cache(allow_output_mutation=True)
27
  def ssr():
28
  ''' Loading ssr model using pipeline api '''