Spaces:
Runtime error
Runtime error
Update function.py
Browse files- 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 '''
|