Update utils.py

utils.py

@@ -28,11 +28,22 @@ def get_res(model_name, input_sentence, single_print=True):
                 pre_len = len(pre) #0
                 w.append(res[pre_len:])  
             pre = res
-                
-        w = [ f'<span style="font-size:1.25em;background-color:{next(color_iterator)}">{x}</span>' for x in w ]
+
+        res = []
+        for x in w:
+            if x == '\n':
+                res.append(x)
+            else:
+                res.append(f'<span style="font-size:1.25em;background-color:{next(color_iterator)}">{x}</span>')
     else:
-        w = [ f'<span style="font-size:1.25em;background-color:{next(color_iterator)}">{tokenizer.decode(x)}</span>' for x in out ]
-    res = ''.join(w)
+        res = []
+        for x in out:
+            if x == '\n':
+                res.append(x)
+            else:
+                res.append(f'<span style="font-size:1.25em;background-color:{next(color_iterator)}">{tokenizer.decode(x)}</span>')
+        
+    res = ''.join(res)
     if single_print:
         print(res + str(token_num))
     else: