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)