wallacelw commited on
Commit
b0ebba5
·
verified ·
1 Parent(s): ddc06e7

Update README.md

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