Downloads · 30 days
9
2% of all-time downloads
electra-pt-br-small-discriminator
josu/electra-pt-br-small-discriminator
electra-pt-br-small-discriminator is a machine learning model from josu. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers.
All-time downloads
389
Public
Repo size
109 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin54.3 MB · 99%
At a glance
- Library
- transformers
- Model type
- electra
- Access
- Public
- Created
- Jan 16, 2023
- Updated
- Jan 16, 2023
- SHA
- 025578e9
Model card
From the Hugging Face model README
from transformers import ElectraForPreTraining, ElectraTokenizerFast
import torch
discriminator = ElectraForPreTraining.from_pretrained("josu/electra-pt-br-small-discriminator")
tokenizer = ElectraTokenizerFast.from_pretrained("josu/electra-pt-br-small-discriminator")
sentence = "os passaros estão cantando"
fake_sentence = "os passaros estão falando"
fake_tokens = tokenizer.tokenize(fake_sentence)
fake_inputs = tokenizer.encode(fake_sentence, return_tensors="pt")
discriminator_outputs = discriminator(fake_inputs)
predictions = torch.round((torch.sign(discriminator_outputs[0]) + 1) / 2)
[print("%7s" % token, end="") for token in fake_tokens]
[print("%7s" % int(prediction), end="") for prediction in predictions.squeeze().tolist()
- Library
- transformers
- Type
- electra
- Languages
- pt
- Created
- Jan 16, 2023
- Updated
- Jan 16, 2023