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 = """
|