dawng88 commited on
Commit
0d9ea28
1 Parent(s): 8099655

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,8 +1,10 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
- pipe = pipeline('sentiment-analysis')
5
  text = st.text_area('enter some text:')
 
 
 
6
 
7
  if text:
8
  out = pipe(text)
 
1
  import streamlit as st
2
  from transformers import pipeline
3
 
 
4
  text = st.text_area('enter some text:')
5
+ classifier = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base", return_all_scores=True)
6
+ classifier(text)
7
+
8
 
9
  if text:
10
  out = pipe(text)