Spaces:
Runtime error
Runtime error
ruchitoshniwal
commited on
Commit
•
b1c2955
1
Parent(s):
f96faeb
Update app.py
Browse files
app.py
CHANGED
@@ -36,7 +36,7 @@ def main():
|
|
36 |
with st.form(key='emotion_detection_form'):
|
37 |
raw_text = st.text_area("Type here")
|
38 |
submit_text = st.form_submit_button(label = 'Submit')
|
39 |
-
st.subheader("
|
40 |
if submit_text:
|
41 |
col1,col2 = st.columns(2)
|
42 |
prediction = predict_emotions(raw_text)
|
@@ -60,7 +60,7 @@ def main():
|
|
60 |
|
61 |
st.markdown("***")
|
62 |
|
63 |
-
st.subheader("
|
64 |
|
65 |
col3,col4 = st.columns(2)
|
66 |
emotion_dataframe_emoroberta, predicted_emotion_emoroberta, prediction_probability_emoroberta = predict_emotions_and_score_emoroberta(raw_text)
|
|
|
36 |
with st.form(key='emotion_detection_form'):
|
37 |
raw_text = st.text_area("Type here")
|
38 |
submit_text = st.form_submit_button(label = 'Submit')
|
39 |
+
st.subheader("Logistic Regression model trained on labeled data (8 emotions)")
|
40 |
if submit_text:
|
41 |
col1,col2 = st.columns(2)
|
42 |
prediction = predict_emotions(raw_text)
|
|
|
60 |
|
61 |
st.markdown("***")
|
62 |
|
63 |
+
st.subheader("EmoRoBERTa - A BERT based pre-trained model fine tuned on reddit comments data(28 emotions)")
|
64 |
|
65 |
col3,col4 = st.columns(2)
|
66 |
emotion_dataframe_emoroberta, predicted_emotion_emoroberta, prediction_probability_emoroberta = predict_emotions_and_score_emoroberta(raw_text)
|