Update README.md

README.md

@@ -15,13 +15,13 @@ base_model:
 ```python
 from transformers import RobertaForMaskedLM, RobertaTokenizer
 
-# Load the pre-trained codeBERT model and tokenizer
-model = RobertaForMaskedLM.from_pretrained("microsoft/codebert-base")
-tokenizer = RobertaTokenizer.from_pretrained("microsoft/codebert-base")
+# Load the pre-trained grammarBERT model and tokenizer
+model = RobertaForMaskedLM.from_pretrained("Nbeau/grammarBERT")
+tokenizer = RobertaTokenizer.from_pretrained("Nbeau/grammarBERT")
 
 # Example of tokenizing a code snippet
 code_snippet = "def enumerate_items(items):"
-derivation_sequence = ast2seq(code_snippet) # ast2seq implementation available https://github.com/NathanaelBeau/grammarBERT/
+derivation_sequence = ast2seq(code_snippet) # ast2seq implementation available https://github.com/NathanaelBeau/grammarBERT/asdl/
 input_ids = tokenizer.encode(code_snippet, return_tensors='pt')
 
 # Predict masked tokens or fine-tune the model as needed