WillWEI0103 commited on
Commit
31890dc
·
verified ·
1 Parent(s): 8f44023

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -1,10 +1,6 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
- def text_summarize(text):
5
- pipe = pipeline("summarization", model="nickmuchi/fb-bart-large-finetuned-trade-the-event-finance-summarizer")
6
- summary = pipe(text)[0]['summary_text']
7
- return summary
8
 
9
  def sentiment(summary):
10
  pipe = pipeline("text-classification", model="WillWEI0103/CustomModel_finance_sentiment_analytics")
@@ -22,7 +18,8 @@ def main():
22
 
23
  #Stage 1: Text Summarization
24
  st.text('Processing Finance News Summarization...')
25
- summary = text_summarize(text)
 
26
  st.write(summary)
27
 
28
  #Stage 2: Sentiment Analytics
 
1
  import streamlit as st
2
  from transformers import pipeline
3
 
 
 
 
 
4
 
5
  def sentiment(summary):
6
  pipe = pipeline("text-classification", model="WillWEI0103/CustomModel_finance_sentiment_analytics")
 
18
 
19
  #Stage 1: Text Summarization
20
  st.text('Processing Finance News Summarization...')
21
+ text_summarize=pipeline("summarization", model="nickmuchi/fb-bart-large-finetuned-trade-the-event-finance-summarizer")
22
+ summary = text_summarize(text)[0]['summary_text']
23
  st.write(summary)
24
 
25
  #Stage 2: Sentiment Analytics