Spaces:

pk248
/

Dialogue-Summarizer

Runtime error

pk248 commited on Dec 17, 2023

Commit

d30219f

•

1 Parent(s): 1c64bbf

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ from transformers import pipeline
 # Function to load the model
-@st.cache(allow_output_mutation=True)
 def load_model():
     summarizer = pipeline("text2text-generation", model="pk248/pegasus-samsum")
     return summarizer

 # Function to load the model
+@st.cache_data(allow_output_mutation=True)
 def load_model():
     summarizer = pipeline("text2text-generation", model="pk248/pegasus-samsum")
     return summarizer