Fix typo in unsqueeze
Browse files
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.
|
| 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 |
```
|