Venustcs commited on
Commit
62a8e80
·
verified ·
1 Parent(s): ec4900f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -1,17 +1,17 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
- # Load the text classification model pipeline
4
  classifier = pipeline("text-classification", model='isom5240sp24/bert-base-uncased-emotion', return_all_scores=True)
5
- # Streamlit application title
6
  st.title("Text Classification for you")
7
  st.write("Classification for 6 emotions: sadness, joy, love, anger, fear, surprise")
8
- # Text input for user to enter the text to classify
9
  text = st.text_area("Enter the text to classify", "")
10
- # Perform text classification when the user clicks the "Classify" button
 
11
  if st.button("Classify"):
12
  results = classifier(text)[0]
13
- # Perform text classification on the input text results = classifier(text)[0]
14
- # Display the classification result
15
  max_score = float('-inf')
16
  max_label = ''
17
  for result in results:
 
1
  import streamlit as st
2
  from transformers import pipeline
3
+
4
  classifier = pipeline("text-classification", model='isom5240sp24/bert-base-uncased-emotion', return_all_scores=True)
5
+
6
  st.title("Text Classification for you")
7
  st.write("Classification for 6 emotions: sadness, joy, love, anger, fear, surprise")
8
+
9
  text = st.text_area("Enter the text to classify", "")
10
+
11
+
12
  if st.button("Classify"):
13
  results = classifier(text)[0]
14
+
 
15
  max_score = float('-inf')
16
  max_label = ''
17
  for result in results: