diff --git a/src/drama_generator/generate.py b/src/drama_generator/generate.py
index d8a0224..46d9b58 100644
--- 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"
", text)
return re.sub(r"\n", r"
\n", text)