Update README.md
Browse files
README.md
CHANGED
|
@@ -15,18 +15,16 @@ metrics:
|
|
| 15 |
- CodeBLEU
|
| 16 |
---
|
| 17 |
|
| 18 |
-
Pretrained model for
|
| 19 |
|
| 20 |
To load the model,
|
| 21 |
|
| 22 |
```
|
| 23 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
| 24 |
-
|
| 25 |
tokenizer = AutoTokenizer.from_pretrained("saikatc/NatGen")
|
| 26 |
-
|
| 27 |
model = AutoModelForSeq2SeqLM.from_pretrained("saikatc/NatGen")
|
| 28 |
```
|
| 29 |
-
|
| 30 |
|
| 31 |
For citation,
|
| 32 |
```
|
|
|
|
| 15 |
- CodeBLEU
|
| 16 |
---
|
| 17 |
|
| 18 |
+
Pretrained model for NatGen: Generative Pre-training by “Naturalizing” Source Code [[`paper`]](https://dl.acm.org/doi/abs/10.1145/3540250.3549162),[[`code`]](https://github.com/saikat107/NatGen),[[`slide`]](https://docs.google.com/presentation/d/1T6kjiohAAR1YvcNvTASR94HptA3xHGCl/edit?usp=sharing&ouid=111755026725574085503&rtpof=true&sd=true).
|
| 19 |
|
| 20 |
To load the model,
|
| 21 |
|
| 22 |
```
|
| 23 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
|
|
|
| 24 |
tokenizer = AutoTokenizer.from_pretrained("saikatc/NatGen")
|
|
|
|
| 25 |
model = AutoModelForSeq2SeqLM.from_pretrained("saikatc/NatGen")
|
| 26 |
```
|
| 27 |
+
|
| 28 |
|
| 29 |
For citation,
|
| 30 |
```
|