Update README.md
Browse files
README.md
CHANGED
@@ -51,8 +51,8 @@ with torch.no_grad():
|
|
51 |
output_scores=True,
|
52 |
)
|
53 |
|
54 |
-
for i in range(len(gen_outputs)):
|
55 |
-
output = tokenizer.decode(gen_outputs[i], skip_special_tokens=True)
|
56 |
print(output)
|
57 |
```
|
58 |
|
|
|
51 |
output_scores=True,
|
52 |
)
|
53 |
|
54 |
+
for i in range(len(gen_outputs[0])):
|
55 |
+
output = tokenizer.decode(gen_outputs[0][i], skip_special_tokens=True)
|
56 |
print(output)
|
57 |
```
|
58 |
|