Update app.py

app.py

@@ -23,12 +23,14 @@ def predict(text):
     issues = [(i, j) for i, j in zipped_list if i.startswith('issue')]
     feelings = [(i, j) for i, j in zipped_list if i.startswith('feeling')]
     harm = [(i, j) for i, j in zipped_list if i.startswith('harm')]
+    sentiment = [(i, j) for i, j in zipped_list if i.startswith('sentiment')]
     # keep tops for each one
     issues = sorted(issues, key=lambda x: x[1])[::-1][:3]
     feelings = sorted(feelings, key=lambda x: x[1])[::-1][:3]
     harm = sorted(harm, key=lambda x: x[1])[::-1][:1]
+    sentiment = sorted(sentiment, key=lambda x: x[1])[::-1][:1]
     # top is the combo of these
-    top = issues + feelings + harm
+    top = issues + feelings + harm + sentiment
 
     logToNotion(text, top)