Update app.py
Browse files
app.py
CHANGED
@@ -5,8 +5,8 @@ from transformers import pipeline
|
|
5 |
classifier = pipeline("text-classification",model='bhadresh-savani/distilbert-base-uncased-emotion', return_all_scores=True)
|
6 |
|
7 |
# Streamlit application title
|
8 |
-
st.title("Text Classification
|
9 |
-
st.write("Classification for 6
|
10 |
|
11 |
# Text input for user to enter the text to classify
|
12 |
text = st.text_area("Enter the text to classify", "")
|
|
|
5 |
classifier = pipeline("text-classification",model='bhadresh-savani/distilbert-base-uncased-emotion', return_all_scores=True)
|
6 |
|
7 |
# Streamlit application title
|
8 |
+
st.title("Text Classification")
|
9 |
+
st.write("Classification for 6 emotions: sadness, joy, love, anger, fear, surprise")
|
10 |
|
11 |
# Text input for user to enter the text to classify
|
12 |
text = st.text_area("Enter the text to classify", "")
|