Update README.md

README.md

@@ -27,8 +27,8 @@ The discriminator should therefore activate the logit corresponding to the fake
 ```python
 from transformers import ElectraForPreTraining, ElectraTokenizerFast
 
-discriminator = ElectraForPreTraining.from_pretrained("Recognai/selectra_small")
-tokenizer = ElectraTokenizerFast.from_pretrained("Recognai/selectra_small")
+discriminator = ElectraForPreTraining.from_pretrained("Recognai/selectra_medium")
+tokenizer = ElectraTokenizerFast.from_pretrained("Recognai/selectra_medium")
 
 sentence_with_fake_token = "Estamos desayunando pan rosa con tomate y aceite de oliva."
 
@@ -39,7 +39,7 @@ 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
+-2.2    -1.9    -6.4    -2.0    -0.6    -4.3    -3.2    -4.9    -5.5    -7.2    -4.5    -4.0
 """
 ```