Cropinky commited on
Commit
0502abd
1 Parent(s): 10788b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -17,8 +17,7 @@ if st.button("Generate lyrics"):
17
  st.title(f"{artist}: {song_name}")
18
  prefix_text = f"<BOS>{song_name} [Verse 1:{artist}]"
19
  generated_song = text_generation(prefix_text, max_length=500, do_sample=True)[0]
20
- for count, line in enumerate(generated_song['generated_text'].split("\
21
- ")):
22
  if count == 0:
23
  st.write(line[line.find('['):])
24
  continue
17
  st.title(f"{artist}: {song_name}")
18
  prefix_text = f"<BOS>{song_name} [Verse 1:{artist}]"
19
  generated_song = text_generation(prefix_text, max_length=500, do_sample=True)[0]
20
+ for count, line in enumerate(generated_song['generated_text'].split("\n")):
 
21
  if count == 0:
22
  st.write(line[line.find('['):])
23
  continue