smangla commited on
Commit
b085fda
verified
1 Parent(s): 7f8fb46

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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))