cosmetic

app.py

@@ -38,9 +38,12 @@ if st.button("Generate lyrics", help="Press me!"):
       if count == 0:
         st.write(line[line.find('['):])
         continue
-      if"<EOS>" in line:
-        break
       if "<BOS>" in line:
         st.write(line[5:])
         continue
+      if line.startswith("["):
+        st.markdown(f"**{line}**")
+        continue
+      if"<EOS>" in line:
+        break
       st.write(line)