namhkoh commited on
Commit
d2f5564
·
1 Parent(s): 9802a55

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +30 -2
README.md CHANGED
@@ -23,7 +23,35 @@ You can use this model directly with a pipeline for masked language modeling:
23
 
24
 
25
  ```python
26
- from transformers import pipeline
27
- unmasker = pipeline('fill-mask', model='MSRoberta')
 
 
 
 
 
 
 
28
  unmasker("Hello, it is a <mask> to meet you.")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
29
  ```
 
23
 
24
 
25
  ```python
26
+ from transformers import pipeline,AutoModelForMaskedLM
27
+ tokenizer = AutoTokenizer.from_pretrained("nkoh01/MSRoberta")
28
+ model = AutoModelForMaskedLM.from_pretrained("nkoh01/MSRoberta")
29
+
30
+ unmasker = pipeline(
31
+ "fill-mask",
32
+ model=model,
33
+ tokenizer=tokenizer
34
+ )
35
  unmasker("Hello, it is a <mask> to meet you.")
36
+
37
+ [{'score': 0.9508683085441589,
38
+ 'sequence': 'hello, it is a pleasure to meet you.',
39
+ 'token': 10483,
40
+ 'token_str': ' pleasure'},
41
+ {'score': 0.015089659951627254,
42
+ 'sequence': 'hello, it is a privilege to meet you.',
43
+ 'token': 9951,
44
+ 'token_str': ' privilege'},
45
+ {'score': 0.013942377641797066,
46
+ 'sequence': 'hello, it is a joy to meet you.',
47
+ 'token': 5823,
48
+ 'token_str': ' joy'},
49
+ {'score': 0.006964420434087515,
50
+ 'sequence': 'hello, it is a delight to meet you.',
51
+ 'token': 13213,
52
+ 'token_str': ' delight'},
53
+ {'score': 0.0024567877408117056,
54
+ 'sequence': 'hello, it is a honour to meet you.',
55
+ 'token': 6671,
56
+ 'token_str': ' honour'}]
57
  ```