Spaces:
Runtime error
Runtime error
File size: 1,306 Bytes
dd25b4b 88055ae dd25b4b 88055ae b34a926 88055ae 758f4ba 652f314 7c7b97f 652f314 88055ae 758f4ba 68de716 88055ae |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
import streamlit as st
from streamlit_player import st_player
from transformers import pipeline
def tester(text):
#classifier = pipeline("sentiment-analysis", arpanghoshal/EmoRoBERTa)
#classifier = pipeline("sentiment-analysis", model='cardiffnlp/twitter-roberta-base-emotion')
#classifier = pipeline("sentiment-analysis", 'j-hartmann/emotion-english-distilroberta-base')
classifier = pipeline("sentiment-analysis", 'cardiffnlp/twitter-roberta-base-emotion')
cardiffnlp/twitter-roberta-base-emotion
results = classifier(text)
st.write(results[0]['label'])
if (results[0]['label']=="anger"):
st_player("https://youtu.be/CmSKVW1v0xM")
elif (results[0]['label']=="disgust"):
st_player("https://youtu.be/CmSKVW1v0xM")
elif (results[0]['label']=="fear"):
st_player("https://youtu.be/CmSKVW1v0xM")
elif (results[0]['label']=="joy"):
st_player("https://youtu.be/CmSKVW1v0xM")
elif (results[0]['label']=="sadness"):
st_player("https://open.spotify.com/playlist/37i9dQZF1E4s6KXygylRFz")
elif (results[0]['label']=="surprise"):
st_player("https://youtu.be/CmSKVW1v0xM")
return results[0]['label']
#return results
emo = st.text_input('Enter text here:')
tester(emo)
|