Update README.md
Browse files
README.md
CHANGED
|
@@ -46,7 +46,7 @@ def predict_answers(batch, model, tokenizer, device):
|
|
| 46 |
|
| 47 |
return [
|
| 48 |
(
|
| 49 |
-
tokenizer.decode(inputs["input_ids"][i][start:end + 1]),
|
| 50 |
(start_probs[i, start] * end_probs[i, end]).item(),
|
| 51 |
)
|
| 52 |
for i, (start, end) in enumerate(zip(start_indices, end_indices))
|
|
|
|
| 46 |
|
| 47 |
return [
|
| 48 |
(
|
| 49 |
+
tokenizer.decode(inputs["input_ids"][i][start:end + 1], skip_special_tokens=True),
|
| 50 |
(start_probs[i, start] * end_probs[i, end]).item(),
|
| 51 |
)
|
| 52 |
for i, (start, end) in enumerate(zip(start_indices, end_indices))
|