From 1156fcdea96838c54023f55eb5f5bf1b14cf6638 Mon Sep 17 00:00:00 2001
From: Tom Weber <tom@weber.codes>
Date: Mon, 7 Nov 2022 13:22:49 +0100
Subject: [PATCH] linebreaks html/md

---
 src/drama_generator/generate.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

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"<br>", text)
     return re.sub(r"\n", r"<br>\n", text)