Update app.py
Browse files
app.py
CHANGED
@@ -11,9 +11,9 @@ def text_summarize(text):
|
|
11 |
return summary
|
12 |
|
13 |
# Story2Audio
|
14 |
-
def sentiment(
|
15 |
pipe = pipeline("text-classification", model="WillWEI0103/CustomModel_finance_sentiment_analytics")
|
16 |
-
label = pipe(
|
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 |
|