Update README.md
Browse files
README.md
CHANGED
|
@@ -40,7 +40,7 @@ masked_index = inputs["input_ids"][0].tolist().index(tokenizer.mask_token_id)
|
|
| 40 |
predicted_token_id = outputs.logits[0, masked_index].argmax(axis=-1)
|
| 41 |
predicted_token = tokenizer.decode(predicted_token_id)
|
| 42 |
print("Predicted token:", predicted_token)
|
| 43 |
-
|
| 44 |
```
|
| 45 |
|
| 46 |
Using a pipeline:
|
|
|
|
| 40 |
predicted_token_id = outputs.logits[0, masked_index].argmax(axis=-1)
|
| 41 |
predicted_token = tokenizer.decode(predicted_token_id)
|
| 42 |
print("Predicted token:", predicted_token)
|
| 43 |
+
|
| 44 |
```
|
| 45 |
|
| 46 |
Using a pipeline:
|