| import transfomers | |
| from transformers import TextClassificationPipeline | |
| pipeline = TextClassificationPipeline( | |
| model=RobertaForSequenceClassification.from_pretrained(CODEBERTA_LANGUAGE_ID), | |
| tokenizer=RobertaTokenizer.from_pretrained(CODEBERTA_LANGUAGE_ID) | |
| ) | |
| pipeline(CODE_TO_IDENTIFY) |