Pranav0111 commited on
Commit
2d7302c
·
verified ·
1 Parent(s): 6c88285

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -51,7 +51,7 @@ with st.spinner("Loading model..."):
51
  tokenizer, model = load_model()
52
 
53
  # Define emotions
54
- emotions = ['anger', 'disgust', 'fear', 'joy', 'neutral', 'sadness', 'surprise']
55
 
56
  # Text input
57
  text_input = st.text_area("Enter your text here:", height=150)
@@ -90,6 +90,5 @@ if st.button("Analyze Emotions"):
90
  # Add footer
91
  st.markdown("---")
92
  st.markdown("""
93
- Created with ❤️ using Hugging Face Transformers and Streamlit.
94
- Model: j-hartmann/emotion-english-distilroberta-base
95
  """)
 
51
  tokenizer, model = load_model()
52
 
53
  # Define emotions
54
+ emotions = ['anger', 'disgust', 'fear', 'joy', 'neutral', 'sadness', 'surprise', 'shock']
55
 
56
  # Text input
57
  text_input = st.text_area("Enter your text here:", height=150)
 
90
  # Add footer
91
  st.markdown("---")
92
  st.markdown("""
93
+ Created by G Pranav Ganesh
 
94
  """)