Spaces:
Sleeping
Sleeping
Update utils.py
Browse files
utils.py
CHANGED
@@ -28,11 +28,22 @@ def get_res(model_name, input_sentence, single_print=True):
|
|
28 |
pre_len = len(pre) #0
|
29 |
w.append(res[pre_len:])
|
30 |
pre = res
|
31 |
-
|
32 |
-
|
|
|
|
|
|
|
|
|
|
|
33 |
else:
|
34 |
-
|
35 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
36 |
if single_print:
|
37 |
print(res + str(token_num))
|
38 |
else:
|
|
|
28 |
pre_len = len(pre) #0
|
29 |
w.append(res[pre_len:])
|
30 |
pre = res
|
31 |
+
|
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:
|
48 |
print(res + str(token_num))
|
49 |
else:
|