Spaces:
Runtime error
Runtime error
Eunchan Lee
commited on
Commit
•
55ec43f
1
Parent(s):
52c1dc8
mlmlmlmlm
Browse files
app.py
CHANGED
@@ -22,7 +22,6 @@ st.image(image, caption='Welcome to KNU Summarizer')
|
|
22 |
|
23 |
plms =["facebook/bart-large-cnn", "google/pegasus-xsum", "t5-small" ]
|
24 |
|
25 |
-
@st.cache(hash_funcs={torch.nn.parameter.Parameter: load_plms})
|
26 |
def load_plms(model_name):
|
27 |
#model_name = "google/pegasus-xsum"
|
28 |
summarizer = pipeline(task="summarization", model=model_name)
|
|
|
22 |
|
23 |
plms =["facebook/bart-large-cnn", "google/pegasus-xsum", "t5-small" ]
|
24 |
|
|
|
25 |
def load_plms(model_name):
|
26 |
#model_name = "google/pegasus-xsum"
|
27 |
summarizer = pipeline(task="summarization", model=model_name)
|