File size: 945 Bytes
6dba858
88055ae
 
6dba858
f301aac
88055ae
cb68822
88055ae
2ecd65e
88055ae
1cd7ed4
6dba858
cb68822
6b072bf
 
9de2735
 
 
6b072bf
4854d0b
d7857b3
 
1cd7ed4
 
d7857b3
1cd7ed4
 
 
 
 
d7857b3
 
1cd7ed4
6b072bf
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
import streamlit.components.v1 as components
from streamlit_player import st_player
from transformers import pipeline
import streamlit as st
import random 


def tester(text):
  classifier = pipeline("sentiment-analysis", model='bhadresh-savani/distilbert-base-uncased-emotion')
  results = classifier(text)
  #st.subheader(results[0]['label'])

#tester(emo)


with st.form(key="form1"):
    emo = st.text_input("Enter a text/phrase/sentence. A corresponding song will be recommended based on its emotion.", placeholder="tester po")
    submit = st.form_submit_button("Generate Playlist!")

tester(emo)
  
  
#def finder(ome):
generator = st.button("Generate Song!")
  
if (generator == True):
  elif (results[0]['label']=="anger"): #songs for anger emotion
    with open('angryplaylist.txt') as f:
      contents = f.read()
    components.html(contents,width=560,height=325)   
  
  
#finder(tester(emo))