Update README.md
Browse files
README.md
CHANGED
@@ -6,7 +6,7 @@ license: mit
|
|
6 |
datasets:
|
7 |
- oscar
|
8 |
widget:
|
9 |
-
- text: "
|
10 |
---
|
11 |
|
12 |
# Indonesian RoBERTa Base
|
@@ -24,7 +24,7 @@ fill_mask = pipeline(
|
|
24 |
tokenizer=pretrained_name
|
25 |
)
|
26 |
|
27 |
-
fill_mask("
|
28 |
```
|
29 |
|
30 |
### Feature Extraction in PyTorch
|
@@ -35,7 +35,7 @@ pretrained_name = "akahana/roberta-base-indonesia"
|
|
35 |
model = RobertaModel.from_pretrained(pretrained_name)
|
36 |
tokenizer = RobertaTokenizerFast.from_pretrained(pretrained_name)
|
37 |
|
38 |
-
prompt = "
|
39 |
encoded_input = tokenizer(prompt, return_tensors='pt')
|
40 |
output = model(**encoded_input)
|
41 |
```
|
|
|
6 |
datasets:
|
7 |
- oscar
|
8 |
widget:
|
9 |
+
- text: "Gajah <mask> sedang makan di kebun binatang."
|
10 |
---
|
11 |
|
12 |
# Indonesian RoBERTa Base
|
|
|
24 |
tokenizer=pretrained_name
|
25 |
)
|
26 |
|
27 |
+
fill_mask("Gajah <mask> sedang makan di kebun binatang.")
|
28 |
```
|
29 |
|
30 |
### Feature Extraction in PyTorch
|
|
|
35 |
model = RobertaModel.from_pretrained(pretrained_name)
|
36 |
tokenizer = RobertaTokenizerFast.from_pretrained(pretrained_name)
|
37 |
|
38 |
+
prompt = "Gajah <mask> sedang makan di kebun binatang."
|
39 |
encoded_input = tokenizer(prompt, return_tensors='pt')
|
40 |
output = model(**encoded_input)
|
41 |
```
|