Update app.py

app.py

@@ -17,15 +17,20 @@ progress_text = "Model loading. Please wait."
 text = st.text_area("Enter your chat", value=predefined_text, height=200)
 button = st.button('Summarize')
 
+def load_pipeline():
+    pipe = pipeline("summarization", model="nicoladisabato/pegasus-samsum")
+    return pipe
+
 if button:
-    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)
+    st.text(progress_text)
+
+    # Load the pipeline with progress visualization
+    with st.spinner("Loading model..."):
+        pipe = st.cache(load_pipeline)()
+
+    st.success("Model loaded.")
+
+    st.text("Generating summary...")
     out = pipe(text)
+
     st.json(out)