Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from transformers import pipeline
|
|
3 |
|
4 |
|
5 |
# Function to load the model
|
6 |
-
@st.
|
7 |
def load_model():
|
8 |
summarizer = pipeline("text2text-generation", model="pk248/pegasus-samsum")
|
9 |
return summarizer
|
|
|
3 |
|
4 |
|
5 |
# Function to load the model
|
6 |
+
@st.cache_data(allow_output_mutation=True)
|
7 |
def load_model():
|
8 |
summarizer = pipeline("text2text-generation", model="pk248/pegasus-samsum")
|
9 |
return summarizer
|