Cropinky commited on
Commit
183cde3
·
1 Parent(s): 295612a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ song_name = st.text_input("Enter the desired song name", "New York Gangstas")
16
  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=1000, 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('['):])
 
16
  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=750, 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('['):])