Update app.py

app.py

@@ -12,13 +12,20 @@ Nico: So won't you come?
 Demo: I prefer to go to Lido Le Canne in Pulsano.
 Nico: Ahahah
 """
-progress_text = "Summarization in progress on CPU. Please wait."
+progress_text = "Model loading. Please wait."
 
 text = st.text_area("Enter your chat", value=predefined_text, height=200)
 button = st.button('Summarize')
-pipe = pipeline("summarization", model="nicoladisabato/pegasus-samsum")
 
 if button:
-    st.progress(0, text=progress_text)
+    st.progress(0)
+
+# Load the pipeline with progress visualization
+    def load_pipeline():
+        pipe = pipeline("summarization", model="nicoladisabato/pegasus-samsum")
+        return pipe
+    
+    pipe = st.cache(load_pipeline)()
+    st.progress(100)
     out = pipe(text)
     st.json(out)