Update README.md

README.md

@@ -28,9 +28,22 @@ We release a `small` and `medium` version with the following configuration:
 ```python
 from transformers import ElectraForPreTraining, ElectraTokenizerFast
 
-discriminator = ElectraForPreTraining.from_pretrained("models/small/pytorch_model")
-tokenizer = ElectraTokenizerFast.from_pretrained("models/medium/pytorch_model")
+discriminator = ElectraForPreTraining.from_pretrained("...")
+tokenizer = ElectraTokenizerFast.from_pretrained("...")
+
+sentence_with_fake_token = "Estamos desayunando pan rosa con tomate y aceite de oliva."
+
+inputs = tokenizer.encode(sentence_with_fake_token, return_tensors="pt")
+logits = discriminator(inputs).logits.tolist()[0]
+
+print("\t".join(tokenizer.tokenize(sentence_with_fake_token)))
+print("\t".join(map(lambda x: str(x)[:4], logits[1:-1])))
+"""Output:
+Estamos desayun ##ando  pan     rosa    con     tomate  y       aceite  de      oliva   .
+-3.1    -3.6    -6.9    -3.0    0.19    -4.5    -3.3    -5.1    -5.7    -7.7    -4.4    -4.2
+"""
 ```
+
 - Links to our zero-shot-classifiers
 
 ## Metrics