Spaces:
Running
Running
Update app.py
Browse files
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
|
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 |
""")
|