NER with electra

Hello Everyone,
I am new to hugging face models. I would like to use electra (electra-large-discriminator-finetuned-conll03-english) for entity recognition. I was unable to find the code to do it. Pointing me in the right direction would be a great help.

Thanks

Hello @swaraj, if I understand correctly you’d like to use Electra for inference right? If yes, then the simplest thing would be to use the pipeline abstraction: Summary of the tasks — transformers 4.3.0 documentation

You can specify the model name to electra-large-discriminator-finetuned-conll03-english which should download the model from the hub and load it in the pipeline for tagging.

1 Like

Thanks a lot @lewtun. Got the stuff. :innocent:

1 Like