Update README.md

README.md

@@ -4,11 +4,13 @@ This is Augmented Code Model which is a fined-tune model of [CodeBERT](https://h
 ## How to use the model ?
 Similar to other huggingface model, you may load the model as follows.
 ```python
-from transformers import AutoTokenizer, AutoModel
 
-tokenizer = AutoTokenizer.from_pretrained("microsoft/codebert-base")
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+
+tokenizer = AutoTokenizer.from_pretrained("Fujitsu/AugCode")
+
+model = AutoModelForSequenceClassification.from_pretrained("Fujitsu/AugCode")
 
-model = AutoModel.from_pretrained("microsoft/codebert-base")
 ```
 Then you may use it to infer the similarity between a given docstring and code.