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 '''
|