devilyouwei commited on
Commit
a364f07
·
verified ·
1 Parent(s): 6855ad8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +15 -0
README.md CHANGED
@@ -59,6 +59,21 @@ training_args = TrainingArguments(
59
 
60
  To train and deploy the SmartBERT V2 model for Web API services, please refer to our GitHub repository: [web3se-lab/SmartBERT](https://github.com/web3se-lab/SmartBERT).
61
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
62
  ## Contributors
63
 
64
  - [Youwei Huang](https://www.devil.ren)
 
59
 
60
  To train and deploy the SmartBERT V2 model for Web API services, please refer to our GitHub repository: [web3se-lab/SmartBERT](https://github.com/web3se-lab/SmartBERT).
61
 
62
+ Or use pipline:
63
+
64
+ ```python
65
+ from transformers import RobertaTokenizer, RobertaForMaskedLM, pipeline
66
+
67
+ model = RobertaForMaskedLM.from_pretrained('web3se/SmartBERT-v3')
68
+ tokenizer = RobertaTokenizer.from_pretrained('web3se/SmartBERT-v3')
69
+
70
+ code_example = "function totalSupply() external view <mask> (uint256);"
71
+ fill_mask = pipeline('fill-mask', model=model, tokenizer=tokenizer)
72
+
73
+ outputs = fill_mask(code_example)
74
+ print(outputs)
75
+ ```
76
+
77
  ## Contributors
78
 
79
  - [Youwei Huang](https://www.devil.ren)