Spaces:
Runtime error
Runtime error
Update function.py
Browse files- function.py +1 -1
function.py
CHANGED
@@ -18,7 +18,7 @@ def bart():
|
|
18 |
''' Loading bart model using pipeline api '''
|
19 |
summarizer = pipeline('summarization',model='facebook/bart-large-cnn')
|
20 |
return summarizer
|
21 |
-
|
22 |
def t5():
|
23 |
''' Loading t5 model using pipeline api '''
|
24 |
summarizer = pipeline("summarization", model="t5-base", tokenizer="t5-base")
|
|
|
18 |
''' Loading bart model using pipeline api '''
|
19 |
summarizer = pipeline('summarization',model='facebook/bart-large-cnn')
|
20 |
return summarizer
|
21 |
+
@st.cache(allow_output_mutation=True)
|
22 |
def t5():
|
23 |
''' Loading t5 model using pipeline api '''
|
24 |
summarizer = pipeline("summarization", model="t5-base", tokenizer="t5-base")
|