Update README.md
Browse files
README.md
CHANGED
|
@@ -76,7 +76,7 @@ masked_index = inputs["input_ids"][0].tolist().index(tokenizer.mask_token_id)
|
|
| 76 |
predicted_token_id = outputs.logits[0, masked_index].argmax(axis=-1)
|
| 77 |
predicted_token = tokenizer.decode(predicted_token_id)
|
| 78 |
print("Predicted token:", predicted_token)
|
| 79 |
-
# Predicted token:
|
| 80 |
```
|
| 81 |
|
| 82 |
Using a pipeline:
|
|
|
|
| 76 |
predicted_token_id = outputs.logits[0, masked_index].argmax(axis=-1)
|
| 77 |
predicted_token = tokenizer.decode(predicted_token_id)
|
| 78 |
print("Predicted token:", predicted_token)
|
| 79 |
+
# Predicted token: Roma
|
| 80 |
```
|
| 81 |
|
| 82 |
Using a pipeline:
|