Spaces:
Runtime error
Runtime error
shawarmabytes
commited on
Commit
·
2ecd65e
1
Parent(s):
0ebbd67
Update app.py
Browse files
app.py
CHANGED
@@ -8,27 +8,26 @@ def tester(text):
|
|
8 |
#classifier = pipeline("sentiment-analysis", model='arpanghoshal/EmoRoBERTa')
|
9 |
#classifier = pipeline("sentiment-analysis", model='cardiffnlp/twitter-roberta-base-emotion')
|
10 |
#classifier = pipeline("sentiment-analysis", 'j-hartmann/emotion-english-distilroberta-base')
|
11 |
-
classifier = pipeline("sentiment-analysis", model='
|
12 |
|
13 |
-
|
14 |
-
|
15 |
results = classifier(text)
|
16 |
st.write(results[0]['label'])
|
17 |
|
18 |
if (results[0]['label']=="anger"):
|
19 |
-
st_player("https://
|
20 |
|
21 |
elif (results[0]['label']=="disgust"):
|
22 |
-
st_player("https://
|
23 |
|
24 |
elif (results[0]['label']=="fear"):
|
25 |
-
st_player("https://
|
26 |
|
27 |
elif (results[0]['label']=="joy"):
|
28 |
-
st_player("https://
|
29 |
|
30 |
elif (results[0]['label']=="sadness"):
|
31 |
-
st_player("https://
|
32 |
|
33 |
elif (results[0]['label']=="surprise"):
|
34 |
st_player("https://youtu.be/CmSKVW1v0xM")
|
@@ -37,7 +36,7 @@ def tester(text):
|
|
37 |
return results[0]['label']
|
38 |
#return results
|
39 |
|
40 |
-
emo = st.text_input('
|
41 |
tester(emo)
|
42 |
|
43 |
|
|
|
8 |
#classifier = pipeline("sentiment-analysis", model='arpanghoshal/EmoRoBERTa')
|
9 |
#classifier = pipeline("sentiment-analysis", model='cardiffnlp/twitter-roberta-base-emotion')
|
10 |
#classifier = pipeline("sentiment-analysis", 'j-hartmann/emotion-english-distilroberta-base')
|
11 |
+
classifier = pipeline("sentiment-analysis", model='bhadresh-savani/distilbert-base-uncased-emotion')
|
12 |
|
13 |
+
|
|
|
14 |
results = classifier(text)
|
15 |
st.write(results[0]['label'])
|
16 |
|
17 |
if (results[0]['label']=="anger"):
|
18 |
+
st_player("https://www.youtube.com/watch?v=kh0BWQ4Uo6w")
|
19 |
|
20 |
elif (results[0]['label']=="disgust"):
|
21 |
+
st_player("https://www.youtube.com/watch?v=zWq2TT3ieGE")
|
22 |
|
23 |
elif (results[0]['label']=="fear"):
|
24 |
+
st_player("https://www.youtube.com/watch?v=iyEUvUcMHgE")
|
25 |
|
26 |
elif (results[0]['label']=="joy"):
|
27 |
+
st_player("https://www.youtube.com/watch?v=1k8craCGpgs")
|
28 |
|
29 |
elif (results[0]['label']=="sadness"):
|
30 |
+
st_player("https://www.youtube.com/watch?v=BZsXcc_tC-oautoplay=1")
|
31 |
|
32 |
elif (results[0]['label']=="surprise"):
|
33 |
st_player("https://youtu.be/CmSKVW1v0xM")
|
|
|
36 |
return results[0]['label']
|
37 |
#return results
|
38 |
|
39 |
+
emo = st.text_input('This application detects the emotion in your text input and suggests a song that matches it. Please enter text below to try:')
|
40 |
tester(emo)
|
41 |
|
42 |
|