Instructions to use dbernsohn/roberta-java with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use dbernsohn/roberta-java with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="dbernsohn/roberta-java")# Load model directly from transformers import AutoTokenizer, AutoModelForMaskedLM tokenizer = AutoTokenizer.from_pretrained("dbernsohn/roberta-java") model = AutoModelForMaskedLM.from_pretrained("dbernsohn/roberta-java") - Notebooks
- Google Colab
- Kaggle
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 = """
|