linebreaks html/md

2 years ago · 1156fcdea9
parent b6bff297f9
commit 1156fcdea9
1 changed files with 1 additions and 3 deletions
--- a/src/drama_generator/generate.py
+++ b/src/drama_generator/generate.py
@ -28,10 +28,8 @@ def load_model(alphabet_size: int, path: str = "saved_models/rnn_2epochs.pth") -
    return gru.eval()


-def make_pretty(text: str, html=False) -> str:
+def make_pretty_md(text: str, html=False) -> str:
    """delete some line breaks for markdown"""
-    if html:
-        return re.sub(r"\n", r"<br>", text)
    return re.sub(r"\n", r"<br>\n", text)