Spaces:
Configuration error
Configuration error
File size: 1,276 Bytes
aade47a |
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 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
import streamlit as st
import get_lyrics
from transformers import pipeline
#driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()))
st.title('Music Lyrics Summarizer')
st.subheader('A Summary of the lyrics of your favourite English songs, prepared by AI')
#lyrics=get_lyrics(inp)
def to_summary(inp):
transcript=''
lyrics=get_lyrics.ask_inp(inp)
for i in lyrics:
transcript+=i+'.'+' '
try:
summed=summarizer(transcript[:len(transcript)],max_length=200,min_length=100)
except IndexError:
summed=summarizer(transcript[:3000],max_length=200,min_length=100)
return summed[0]['summary_text']
summarizer = pipeline("summarization", model="knkarthick/MEETING-SUMMARY-BART-LARGE-XSUM-SAMSUM-DIALOGSUM")
def take_inp():
inp=st.text_input(label='Insert song name and artist name:')
return inp
inp=take_inp()
if len(inp)>2:
st.caption('Please wait while the AI tries to read the lyrics of your song from Google and understand it.')
st.caption("<-> You have a cool music taste. But what's cooler is my ability to understand music <->")
st.write(to_summary(inp))
st.stop()
else:
st.stop()
if __name__=='__main__':
take_inp()
|