haukurpj commited on
Commit
dca38d2
·
verified ·
1 Parent(s): aa55a31

Fix typo in unsqueeze

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -57,7 +57,7 @@ print("IFD conversion test passed successfully!")
57
 
58
  # Alternative: use prepare_inputs for single sentence prediction
59
  input_ids, attention_mask, word_mask = model.prepare_inputs(sentence_words, tokenizer)
60
- single_result = model.predict_labels(input_ids.unsqeeze(0), attention_mask.unsqeeze(0), word_mask.unsqeeze(0))
61
  assert single_result == expected, f"Expected {expected}, but got {single_result}"
62
  print("Single sentence prediction test passed successfully!")
63
  ```
 
57
 
58
  # Alternative: use prepare_inputs for single sentence prediction
59
  input_ids, attention_mask, word_mask = model.prepare_inputs(sentence_words, tokenizer)
60
+ single_result = model.predict_labels(input_ids.unsqueeze(0), attention_mask.unsqueeze(0), word_mask.unsqueeze(0))
61
  assert single_result == expected, f"Expected {expected}, but got {single_result}"
62
  print("Single sentence prediction test passed successfully!")
63
  ```