Added Files and Information

README.md

@@ -1,10 +1,10 @@
 ---
-title: Song Title Generation
-emoji: 🦀
-colorFrom: pink
-colorTo: yellow
+title: Genius Summarization Space
+emoji: ⚡
+colorFrom: indigo
+colorTo: indigo
 sdk: gradio
-sdk_version: 3.41.2
+sdk_version: 3.39.0
 app_file: app.py
 pinned: false
 ---

app.py

@@ -0,0 +1,239 @@
+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()

requirements.txt

@@ -0,0 +1,2 @@
+transformers
+torch