Update app.py

app.py

@@ -1,11 +1,11 @@
 import streamlit as st
 from transformers import pipeline
 
-def text_summarize(text):
-    pipe = pipeline("summarization", model="human-centered-summarization/financial-summarization-pegasus")
-    summary = pipe(text)[0]['summary_text']
-    print(summary)
-    return summary
+#def text_summarize(text):
+#    pipe = pipeline("summarization", model="human-centered-summarization/financial-summarization-pegasus")
+#    summary = pipe(text)[0]['summary_text']
+#    print(summary)
+#    return summary
 
 def sentiment(summary):
     pipe = pipeline("text-classification", model="WillWEI0103/CustomModel_finance_sentiment_analytics")
@@ -19,7 +19,9 @@ def main():
     text=st.text_input('Input your Finance news: ')
     
     #Stage 1: Text Summarization
+    text_summarize=pipeline("summarization", model="human-centered-summarization/financial-summarization-pegasus")
     summary = text_summarize(text)
+    summary = summary[0]['summary_text']
     st.text('Processing Finance News Summarization...')
     st.write(summary)