|
|
--- |
|
|
license: mit |
|
|
--- |
|
|
|
|
|
This is an example model trained on C declarations. It is a foundational model that can essentially only complete C declarations. |
|
|
|
|
|
It has been trained on a variety of OSS software declarations, not entire functions. |
|
|
|
|
|
Usage: |
|
|
|
|
|
```python |
|
|
# make a prediction |
|
|
from transformers import pipeline |
|
|
from pprint import pprint |
|
|
|
|
|
fill_mask = pipeline( |
|
|
"fill-mask", |
|
|
model="mstaron/declBERTa", |
|
|
tokenizer="mstaron/declBERTa" |
|
|
) |
|
|
|
|
|
strPredicted = fill_mask("int i = <mask>;", top_k=10) |
|
|
|
|
|
pprint(strPredicted) |
|
|
|
|
|
``` |
