Update app.py

app.py

@@ -2,29 +2,28 @@ import streamlit as st
 from transformers import pipeline
 
 def main():
-    # Initialize the application interface
-    st.title("📈 Financial Text Summarization with Hugging Face Spaces 🚀")
-    st.write("Enter a financial text to generate its summary:")
+    # Streamlit app interface design
+    st.title("📈 Financial Text Summarization")
+    st.write("Please enter the financial text below to generate a concise summary:")
 
-    # User input area
-    user_input = st.text_area("Text Input 📝", height=150)
+    # User input area for text
+    user_input = st.text_area("Text Input 📝", height=300)
 
-    # Load the summarization pipeline
-    # Ensuring the pipeline is loaded inside the function to prevent re-loading on every interaction
+    # Initialize the summarization pipeline only once using st.session_state
     if 'summarizer' not in st.session_state:
-        st.session_state['summarizer'] = pipeline("summarization", model="human-centered-summarization/financial-summarization-pegasus")
+        st.session_state.summarizer = pipeline("summarization", model="human-centered-summarization/financial-summarization-pegasus")
 
-    # Button to trigger summarization
+    # Button to generate summary
     if st.button("Generate Summary"):
         if user_input:
-            # Generating summary
             with st.spinner('Generating summary...'):
-                summary_result = st.session_state['summarizer'](user_input, max_length=130, min_length=30, do_sample=False)
+                # Generating the summary from the input text
+                summary_result = st.session_state.summarizer(user_input, max_length=130, min_length=30, do_sample=False)
                 summary = summary_result[0]['summary_text']
                 st.write("Summary 📜:")
                 st.write(summary)
         else:
-            st.error("Please enter some text to summarize.")
+            st.error("Please enter some text to generate a summary.")
 
 if __name__ == "__main__":
     main()