WillWEI0103 commited on
Commit
2e23545
·
verified ·
1 Parent(s): c31d7ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,9 +11,9 @@ def text_summarize(text):
11
  return summary
12
 
13
  # Story2Audio
14
- def sentiment(story_text):
15
  pipe = pipeline("text-classification", model="WillWEI0103/CustomModel_finance_sentiment_analytics")
16
- label = pipe(story_text)[0]['label']
17
  return label
18
 
19
 
 
11
  return summary
12
 
13
  # Story2Audio
14
+ def sentiment(summary):
15
  pipe = pipeline("text-classification", model="WillWEI0103/CustomModel_finance_sentiment_analytics")
16
+ label = pipe(summary)[0]['label']
17
  return label
18
 
19