Daulet9900 commited on
Commit
8e632d2
·
1 Parent(s): 3f1920d

Add model "bhadresh-savani/distilbert-base-uncased-emotion"

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -3,10 +3,11 @@ from transformers import pipeline
3
 
4
  print ("Load hv model...")
5
 
6
- #Title and Description
7
- st.title("Title to space")
8
- st.write("""This will be spaces description(manual)
9
- """)
10
 
11
- x = st.slider('Select a value')
12
- st.write(x, 'squared is', x * x)
 
 
 
3
 
4
  print ("Load hv model...")
5
 
6
+ # Load the pre-trained emotion classification pipeline
7
+ model_name = "bhadresh-savani/distilbert-base-uncased-emotion"
8
+ emotion_classifier = pipeline("text-classification", model=model_name)
 
9
 
10
+ # Title and Description
11
+ st.title("Emotion Classifier")
12
+ st.write("""write down how your day went or what your mood is.
13
+ """)