import gradio as gr from transformers import pipeline pipe = pipeline("summarization", model="miscjose/mt5-small-finetuned-genius-music") def predict(Lyrics): return pipe(Lyrics)[0]['summary_text'] examples = [[ ''' Thought I'd end up with Sean But he wasn't a match Wrote some songs about Ricky Now I listen and laugh Even almost got married And for Pete, I'm so thankful Wish I could say, "Thank you" to Malcolm 'Cause he was an angel One taught me love One taught me patience And one taught me pain Now, I'm so amazing Say I've loved and I've lost But that's not what I see So, look what I got Look what you taught me And for that, I say Thank you, next (Next) Thank you, next (Next) Thank you, next I'm so fuckin' grateful for my ex Thank you, next (Next) Thank you, next (Next) Thank you, next (Next) I'm so fuckin'— Spend more time with my friends I ain't worried 'bout nothin' Plus, I met someone else We havin' better discussions I know they say I move on too fast But this one gon' last 'Cause her name is Ari And I'm so good with that (So good with that) She taught me love (Love) She taught me patience (Patience) How she handles pain (Pain) That shit's amazing (Yeah, she's amazing) I've loved and I've lost (Yeah, yeah) But that's not what I see (Yeah, yeah) 'Cause look what I've found (Yeah, yeah, I've found) Ain't no need for searching, and for that, I say Thank you, next (Thank you, next) Thank you, next (Thank you, next) Thank you, next (Thank you) I'm so fuckin' grateful for my ex Thank you, next (Thank you, next) Thank you, next (Said thank you, next) Thank you, next (Next) I'm so fuckin' grateful for my ex Thank you, next Thank you, next Thank you, next I'm so fuckin'— One day I'll walk down the aisle Holding hands with my mama I'll be thanking my dad 'Cause she grew from the drama Only wanna do it once, real bad Gon' make that shit last God forbid something happens Least this song is a smash (Song is a smash) I've got so much love (Love) Got so much patience (Patience) And I've learned from the pain (Pain) I turned out amazing (Turned out amazing) Say I've loved and I've lost (Yeah, yeah) But that's not what I see (Yeah, yeah) 'Cause look what I've found (Yeah, yeah) Ain't no need for searching And for that, I say Thank you, next (Thank you, next) Thank you, next (Thank you, next) Thank you, next I'm so fuckin' grateful for my ex Thank you, next (Thank you, next) Thank you, next (Said thank you, next) Thank you, next (Next) I'm so fuckin' grateful for my ex Thank you, next Thank you, next Thank you, next Yeah, yee Thank you, next Thank you, next Thank you, next Yeah, yee '''], ['''Who's this? Da-ddy Yan-kee! Zúmbale mambo pa' que mis gatas prendan los motore' Zúmbale mambo pa' que mis gatas prendan los motore' Zúmbale mambo pa' que mis gatas prendan los motore' Que se preparen que lo que viene es pa' que le den ¡Duro! Mamita, yo sé que tú no te me va' a quitar ¡Duro! Lo que me gusta es que tú te dejas llevar ¡Duro! Todos los weekend'es ella sale a vacilar ¡Duro! Mi gata no para 'e janguear, porque A ella le gusta la gasolina (Dame más gasolina) Cómo le encanta la gasolina (Dame más gasolina) A ella le gusta la gasolina (Dame más gasolina) Cómo le encanta la gasolina (Dame más gasolina) Ella prende las turbinas, no discrimina No se pierde ni un party de marquesina Se acicala hasta pa' la esquina Luce tan bien que hasta la sombra le combina Asesina, me domina Janguea en carros, motoras y limusinas Llena su tanque de adrenalina Cuando escucha reggaetón en las bocinas A ella le gusta la gasolina (Dame más gasolina) Cómo le encanta la gasolina (Dame más gasolina) A ella le gusta la gasolina (Dame más gasolina) Cómo le encanta la gasolina (Dame más gasolina) Aquí somos los mejores, no te me ajore' En la pista nos llaman "Los Matadore'" Tú hace' que cualquiera se enamore Cuando bailas al ritmo de los tambore' Esto va pa' las gatas de to's colore' Pa' las mayore', pa' las menore' Pa' las que son más zorras que los cazadore' Pa' las mujeres que no apagan sus motore' Tenemo' tú y yo algo pendiente Tú me debes algo y lo sabe' Conmigo ella se pierde No le rinde cuentas a nadie Tenemo' tú y yo algo pendiente Tú me debes algo y lo sabe' Conmigo ella se pierde No le rinde cuentas a nadie Zúmbale mambo pa' que mis gatas prendan los motore' Zúmbale mambo pa' que mis gatas prendan los motore' Zúmbale mambo pa' que mis gatas prendan los motore' Que se preparen que lo que viene es pa' que le den ¡Duro! Mamita, yo sé que tú no te me va' a quitar ¡Duro! Lo que me gusta es que tú te dejas llevar ¡Duro! Todos los weekend'es ella sale a vacilar ¡Duro! Mi gata no para 'e janguear, porque A ella le gusta la gasolina (Dame más gasolina) Cómo le encanta la gasolina (Dame más gasolina) A ella le gusta la gasolina (Dame más gasolina) Cómo le encanta la gasolina (Dame más gasolina) '''], [''' 無敵の笑顔で荒らすメディア 知りたいその秘密ミステリアス 抜けてるとこさえ彼女のエリア 完璧で嘘つきな君は 天才的なアイドル様 今日何食べた？ 好きな本は？ 遊びに行くならどこに行くの？ 何も食べてない それは内緒 何を聞かれても のらりくらり そう淡々と だけど燦々と 見えそうで見えない秘密は蜜の味 あれもないないない これもないないない 好きなタイプは？ 相手は？ さあ答えて 「誰かを好きになることなんて私分からなくてさ」 嘘か本当か知り得ない そんな言葉にまた一人堕ちる また好きにさせる 誰もが目を奪われていく 君は完璧で究極のアイドル 金輪際現れない 一番星の生まれ変わり その笑顔で愛してるで 誰も彼も虜にしていく その瞳がその言葉が 嘘でもそれは完全なアイ はいはいあの子は特別です 我々はハナからおまけです お星様の引き立て役Bです 全てがあの子のお陰なわけない 洒落臭い 妬み嫉妬なんてないわけがない これはネタじゃない からこそ許せない 完璧じゃない君じゃ許せない 自分を許せない 誰よりも強い君以外は認めない 誰もが信じ崇めてる まさに最強で無敵のアイドル 弱点なんて見当たらない 一番星を宿している 弱いとこなんて見せちゃダメダメ 知りたくないとこは見せずに 唯一無二じゃなくちゃイヤイヤ それこそ本物のアイ 得意の笑顔で沸かすメディア 隠しきるこの秘密だけは 愛してるって嘘で積むキャリア これこそ私なりの愛だ 流れる汗も綺麗なアクア ルビーを隠したこの瞼 歌い踊り舞う私はマリア そう嘘はとびきりの愛だ 誰かに愛されたことも 誰かのこと愛したこともない そんな私の嘘がいつか本当になること 信じてる いつかきっと全部手に入れる 私はそう欲張りなアイドル 等身大でみんなのこと ちゃんと愛したいから 今日も嘘をつくの この言葉がいつか本当になる日を願って それでもまだ 君と君にだけは言えずにいたけど やっと言えた これは絶対嘘じゃない 愛してる ''']] interface = gr.Interface( title='Song Title Generation (mt5-small-finetuned-genius-music)', description="Enter song lyrics to generate song titles! Click the examples bellow to view the model's inference of Ariana Grande's 'thank u next', Daddy Yankee's 'Gasolina', and YOASOBI's 'アイドル'", fn=predict, inputs=gr.Textbox(lines=7, placeholder="text ...", label='Song Lyrics'), outputs=gr.Textbox(lines=1, label='Predicted Title'), examples=examples ) interface.launch()