Update app.py
Browse files
app.py
CHANGED
|
@@ -70,6 +70,17 @@ def sentence_cleaning(sentence):
|
|
| 70 |
|
| 71 |
|
| 72 |
def main():
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 73 |
st.title("Emotion Classifier")
|
| 74 |
menu = ["Home", "Monitor"]
|
| 75 |
choice = st.sidebar.selectbox("Menu", menu)
|
|
@@ -91,6 +102,7 @@ def main():
|
|
| 91 |
st.write(raw_text)
|
| 92 |
|
| 93 |
st.success("Prediction")
|
|
|
|
| 94 |
st.write("Confidence:{}".format(proba))
|
| 95 |
|
| 96 |
|
|
|
|
| 70 |
|
| 71 |
|
| 72 |
def main():
|
| 73 |
+
emoji = { "anger":"😠",
|
| 74 |
+
"disgust":"🤮",
|
| 75 |
+
"fear":"😨😱",
|
| 76 |
+
"happy":"🤗",
|
| 77 |
+
"joy":"😂",
|
| 78 |
+
"neutral":"😐",
|
| 79 |
+
"sad":"😔",
|
| 80 |
+
"sadness":"😔",
|
| 81 |
+
"shame":"😳",
|
| 82 |
+
"surprise":"😮"
|
| 83 |
+
}
|
| 84 |
st.title("Emotion Classifier")
|
| 85 |
menu = ["Home", "Monitor"]
|
| 86 |
choice = st.sidebar.selectbox("Menu", menu)
|
|
|
|
| 102 |
st.write(raw_text)
|
| 103 |
|
| 104 |
st.success("Prediction")
|
| 105 |
+
st.write("{}:{}".format(res, emoji[res]))
|
| 106 |
st.write("Confidence:{}".format(proba))
|
| 107 |
|
| 108 |
|