Update README.md

README.md

@@ -33,8 +33,6 @@ A BERT-like model pre-trained on Java buggy code.
 
 Fill-Mask.
 
-[More Information Needed]
-
 ### Downstream Use [optional]
 
 The model can be used for other tasks, like Text Classification.
@@ -63,7 +61,7 @@ Use the code below to get started with the model.
 
 ```python
 from transformers import pipeline
-unmasker = pipeline('fill-mask', model='bert-base-cased')
+unmasker = pipeline('fill-mask', model='bert-java-bfp_single')
 unmasker(java_code) # Replace with Java code; Use '[MASK]' to mask tokens/words in the code.
 ```
 
@@ -81,7 +79,7 @@ The model was trained on 236040 Java methods, containing the code before and aft
 
 #### Preprocessing [optional]
 
-Remove comments and replace consecutive whitespace characters by a single space
+Remove comments and replace consecutive whitespace characters by a single space.
 
 #### Training Hyperparameters