Cropinky commited on
Commit
be51355
·
1 Parent(s): f43cb7c
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ if st.button("Generate lyrics", help="Press me!"):
36
  generated_song = text_generation(prefix_text, max_length=750, do_sample=True)[0]
37
  for count, line in enumerate(generated_song['generated_text'].split("\n")):
38
  if count == 0:
39
- st.write(line[line.find('['):])
40
  continue
41
  if "<BOS>" in line:
42
  st.write(line[5:])
 
36
  generated_song = text_generation(prefix_text, max_length=750, do_sample=True)[0]
37
  for count, line in enumerate(generated_song['generated_text'].split("\n")):
38
  if count == 0:
39
+ st.markdown(f"**{line[line.find('['):]}**")
40
  continue
41
  if "<BOS>" in line:
42
  st.write(line[5:])