Annorita commited on
Commit
aa0404d
1 Parent(s): 8de26e7

Update utils.py

Browse files

print space in markdown

Files changed (1) hide show
  1. utils.py +4 -4
utils.py CHANGED
@@ -32,16 +32,16 @@ def get_res(model_name, input_sentence, single_print=True):
32
  res = []
33
  for x in w:
34
  if x == '\n':
35
- res.append(x)
36
  else:
37
- res.append(f'<span style="font-size:1.25em;background-color:{next(color_iterator)}">{x}</span>')
38
  else:
39
  res = []
40
  for x in out:
41
  if x == '\n':
42
- res.append(x)
43
  else:
44
- res.append(f'<span style="font-size:1.25em;background-color:{next(color_iterator)}">{tokenizer.decode(x)}</span>')
45
 
46
  res = ''.join(res)
47
  if single_print:
 
32
  res = []
33
  for x in w:
34
  if x == '\n':
35
+ res.append(' \n')
36
  else:
37
+ res.append(f'<span style="font-size:1.25em;background-color:{next(color_iterator)}">{x.replace(" ", "&nbsp;")}</span>')
38
  else:
39
  res = []
40
  for x in out:
41
  if x == '\n':
42
+ res.append(' \n')
43
  else:
44
+ res.append(f'<span style="font-size:1.25em;background-color:{next(color_iterator)}">{tokenizer.decode(x).replace(" ", "&nbsp;")}</span>')
45
 
46
  res = ''.join(res)
47
  if single_print: