Spaces:
Runtime error
Runtime error
Update function.py
Browse files- function.py +1 -1
function.py
CHANGED
@@ -23,7 +23,7 @@ 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="santiviquez/ssr-base-finetuned-samsum-en")
|
|
|
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 bart_t5():
|
28 |
''' Loading bart_t5 model using pipeline api '''
|
29 |
summarizer = pipeline("summarization", model="santiviquez/ssr-base-finetuned-samsum-en")
|