Modelpublic

addy88/programming-lang-identifier

sourceHugging Faceupdated 5y agoView on Hugging Face

7likes32downloads

Model Card

This model is funetune version of Codebert in roberta. On CodeSearchNet. ### Quick start:

from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("addy88/programming-lang-identifier")

model = AutoModelForSequenceClassification.from_pretrained("addy88/programming-lang-identifier")

inputids = tokenizer.encode(CODETOIDENTIFY) logits = model(inputids)[0]

language_idx = logits.argmax() # index for the resulting label ###