sss

app.py

@@ -18,7 +18,7 @@ model_name_xsum = 'google/pegasus-xsum'
 tokenizer = AutoTokenizer.from_pretrained(model_name_xsum)
 model = PegasusForConditionalGeneration.from_pretrained(model_name_xsum).to(device)
 
-
+@st.cache
 def pegasus_abs_summarize(src_text):
         
     
@@ -35,4 +35,3 @@ st.text_input("Input:", key="input")
 
 abs_output = pegasus_abs_summarize(st.session_state.input)
 
-st.write('Abstractive output', abs_output)