Update utils.py

utils.py

@@ -29,7 +29,7 @@ def get_res(model_name, input_sentence, single_print=True):
                 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 out ]
+        w = [ f'<span style="font-size:1.25em;background-color:{next(color_iterator)}">{x}</span>' for x in w ]
     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)