Update README.md

README.md

@@ -42,15 +42,13 @@ This is the model card of a 🤗 transformers model that has been pushed on the
 
 <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
 ```python
-from transformers import pipeline
+from transformers import pipeline, TextClassificationPipeline, AutoTokenizer, AutoModelForSequenceClassification
 
 checkpoint = "malteklaes/based-CodeBERTa-language-id-llm-module"
 
-from transformers import TextClassificationPipeline
-
 myPipeline = TextClassificationPipeline(
-    model=RobertaForSequenceClassification.from_pretrained(checkpoint),
-    tokenizer=RobertaTokenizer.from_pretrained(checkpoint)
+    model=AutoModelForSequenceClassification.from_pretrained(checkpoint),
+    tokenizer=AutoTokenizer.from_pretrained(checkpoint)
 )
 
 CODE_TO_IDENTIFY = "print('hello world')"