MatthewsO3 commited on
Commit
1710164
·
verified ·
1 Parent(s): 4805afc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -48,7 +48,7 @@ This model can be used as-is for masked token prediction in C++ and Erlang codeb
48
  ```python
49
  from transformers import pipeline
50
 
51
- fill = pipeline("fill-mask", model="<your-model-id>")
52
  fill("int <mask> = 0;")
53
  ```
54
 
@@ -84,7 +84,7 @@ from transformers import RobertaTokenizer, RobertaForMaskedLM
84
  import torch
85
 
86
  tokenizer = RobertaTokenizer.from_pretrained("microsoft/graphcodebert-base")
87
- model = RobertaForMaskedLM.from_pretrained("<your-model-id>")
88
 
89
  code = "int <mask> = 42;"
90
  inputs = tokenizer(code, return_tensors="pt")
 
48
  ```python
49
  from transformers import pipeline
50
 
51
+ fill = pipeline("fill-mask", model="MatthewsO3/GraphCode-CErl-base")
52
  fill("int <mask> = 0;")
53
  ```
54
 
 
84
  import torch
85
 
86
  tokenizer = RobertaTokenizer.from_pretrained("microsoft/graphcodebert-base")
87
+ model = RobertaForMaskedLM.from_pretrained("MatthewsO3/GraphCode-CErl-base")
88
 
89
  code = "int <mask> = 42;"
90
  inputs = tokenizer(code, return_tensors="pt")