WillWEI0103 commited on
Commit
a479b7a
·
verified ·
1 Parent(s): 03855c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -19,7 +19,8 @@ def main():
19
  text=st.text_input('Input your Finance news: ')
20
 
21
  #Stage 1: Text Summarization
22
- text_summarize=pipeline("summarization", model="human-centered-summarization/financial-summarization-pegasus")
 
23
  summary = text_summarize(text)
24
  summary = summary[0]['summary_text']
25
  st.text('Processing Finance News Summarization...')
 
19
  text=st.text_input('Input your Finance news: ')
20
 
21
  #Stage 1: Text Summarization
22
+ text_summarize=pipeline("summarization", model="nickmuchi/fb-bart-large-finetuned-trade-the-event-finance-summarizer")
23
+ #text_summarize=pipeline("summarization", model="human-centered-summarization/financial-summarization-pegasus")
24
  summary = text_summarize(text)
25
  summary = summary[0]['summary_text']
26
  st.text('Processing Finance News Summarization...')