Update app.py

app.py

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