Update README.md
Browse files
README.md
CHANGED
|
@@ -11,8 +11,8 @@ To load the model:
|
|
| 11 |
(necessary packages: !pip install transformers sentencepiece)
|
| 12 |
```python
|
| 13 |
from transformers import AutoTokenizer, AutoModelWithLMHead, pipeline
|
| 14 |
-
tokenizer = AutoTokenizer.from_pretrained("dbernsohn/roberta-
|
| 15 |
-
model = AutoModelWithLMHead.from_pretrained("dbernsohn/roberta-
|
| 16 |
|
| 17 |
fill_mask = pipeline(
|
| 18 |
"fill-mask",
|
|
@@ -21,7 +21,7 @@ fill_mask = pipeline(
|
|
| 21 |
)
|
| 22 |
```
|
| 23 |
|
| 24 |
-
You can then use this model to fill masked words in a
|
| 25 |
|
| 26 |
```java
|
| 27 |
code = """
|
|
|
|
| 11 |
(necessary packages: !pip install transformers sentencepiece)
|
| 12 |
```python
|
| 13 |
from transformers import AutoTokenizer, AutoModelWithLMHead, pipeline
|
| 14 |
+
tokenizer = AutoTokenizer.from_pretrained("dbernsohn/roberta-java")
|
| 15 |
+
model = AutoModelWithLMHead.from_pretrained("dbernsohn/roberta-java")
|
| 16 |
|
| 17 |
fill_mask = pipeline(
|
| 18 |
"fill-mask",
|
|
|
|
| 21 |
)
|
| 22 |
```
|
| 23 |
|
| 24 |
+
You can then use this model to fill masked words in a Java code.
|
| 25 |
|
| 26 |
```java
|
| 27 |
code = """
|