dbernsohn commited on
Commit
1061bc9
·
1 Parent(s): 075f3cb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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-python")
15
- model = AutoModelWithLMHead.from_pretrained("dbernsohn/roberta-python")
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 Pytho code.
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 = """