Spaces:
Runtime error
Runtime error
Update function.py
Browse files- function.py +2 -2
function.py
CHANGED
@@ -24,8 +24,8 @@ def t5():
|
|
24 |
summarizer = pipeline("summarization", model="t5-base", tokenizer="t5-base")
|
25 |
return summarizer
|
26 |
@st.cache(allow_output_mutation=True)
|
27 |
-
def
|
28 |
-
''' Loading
|
29 |
summarizer = pipeline("summarization", model="santiviquez/ssr-base-finetuned-samsum-en")
|
30 |
return summarizer
|
31 |
|
|
|
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 '''
|
29 |
summarizer = pipeline("summarization", model="santiviquez/ssr-base-finetuned-samsum-en")
|
30 |
return summarizer
|
31 |
|