radio

app.py

@@ -10,17 +10,26 @@ def load_model():
   model = AutoModelForCausalLM.from_pretrained(model_ckpt,from_flax=True)
   return tokenizer, model
 
+@st.cache()
+def load_rappers():
+  text_file = open("rappers.txt")
+  rappers = text_file.readlines()
+  rappers.sort()
+  return rappers
+
 
 title = st.title("Loading model")
 tokenizer, model = load_model()
 text_generation = pipeline("text-generation", model=model, tokenizer=tokenizer)
 title.title("Rap lyrics generator")
-artist = st.text_input("Enter the artist", "Wu-Tang Clan")
-song_name = st.text_input("Enter the desired song name", "New York Gangstas")
+#artist = st.text_input("Enter the artist", "Wu-Tang Clan")
+list_of_rappers = load_rappers()
+artist = st.radio("Choose your rapper", tuple(list_of_rappers))
+song_name = st.text_input("Enter the desired song name", "Shaolin")
 
 
 
-if st.button("Generate lyrics"):
+if st.button("Generate lyrics", help="Press me!"):
     st.title(f"{artist}: {song_name}")
     prefix_text = f"<BOS>{song_name} [Verse 1:{artist}]"
     generated_song = text_generation(prefix_text, max_length=750, do_sample=True)[0]

rappers.txt

@@ -0,0 +1,53 @@
+The Notorious B.I.G. 
+Big Pun
+Big L
+Nas
+50 Cent
+Prodigy
+Action Bronson
+Ill Bill
+Lil Keed
+Lil Keke
+Lil Jon
+Lil Bibby
+Lil Tjay
+Jay Z
+Guru
+Meek Mill
+Yung Lean
+Bladee
+Tupac
+Snoop Dogg
+Eminem
+Vince Staples
+Rakim
+Big daddy kane
+Kendrick Lamar
+Big Boi
+Danny Brown
+Freddie Gibbs
+Tyler the creator
+Frank Ocean
+Earl Eweatshirt
+Domo Genesis
+Mike G
+Raekwon
+Ghostface Killah
+RZA
+GZA
+Ol' Dirty Bastard
+Method Man
+Inspectah Deck
+U-God
+Masta Killa
+Cappadonna
+Kanye West
+Lauryn Hill
+Jean Grae
+Lil Kim
+Missy Elliot
+Rah Digga
+MC Lyte
+Remy Ma
+Missy Elliott
+Foxy Brown