Update app.py

app.py

@@ -8,7 +8,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_ckpt,from_flax=True)
 model = AutoModelForCausalLM.from_pretrained(model_ckpt,from_flax=True)
 text_generation = pipeline("text-generation", model=model, tokenizer=tokenizer)
 title.title("Rap lyrics generator")
-artist = st.text_input("Enter the artist", "Eminem")
+artist = st.text_input("Enter the artist", "Jay-Z")
 song_name = st.text_input("Enter the desired song name", "Gas is going")
 
 
@@ -17,7 +17,8 @@ if st.button("Generate lyrics"):
     st.title(f"{artist}: {song_name}")
     prefix_text = f"<BOS>{song_name} [Verse 1:{artist}]"
     generated_song = text_generation(prefix_text, max_length=500, do_sample=True)[0]
-    for count, line in enumerate(generated_song['generated_text'].split("\n")):
+    for count, line in enumerate(generated_song['generated_text'].split("\
+")):
       if count == 0:
         st.write(line[line.find('['):])
         continue