Mintiny commited on
Commit
7157357
1 Parent(s): 81a574e

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -30
app.py DELETED
@@ -1,30 +0,0 @@
1
- import streamlit as st
2
- from transformers import pipeline
3
-
4
- # Load the text classification model pipeline
5
- classifier = pipeline("text-classification",model='isom5240sp24/bert-base-uncased-emotion', return_all_scores=True)
6
-
7
- # Streamlit application title
8
- st.title("Text Classification for you")
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", "")
13
-
14
- # Perform text classification when the user clicks the "Classify" button
15
- if st.button("Classify"):
16
- # Perform text classification on the input text
17
- results = classifier(text)[0]
18
-
19
- # Display the classification result
20
- max_score = float('-inf')
21
- max_label = ''
22
-
23
- for result in results:
24
- if result['score'] > max_score:
25
- max_score = result['score']
26
- max_label = result['label']
27
-
28
- st.write("Text:", text)
29
- st.write("Label:", max_label)
30
- st.write("Score:", max_score)