ruchitoshniwal commited on
Commit
b1c2955
1 Parent(s): f96faeb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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("Predictions from LR model trained on labeled data (8 emotions)")
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("Predictions from EmoRoBERTa - BERT based pre-trained model (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)
 
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)