Update README.md
Browse files
README.md
CHANGED
|
@@ -32,7 +32,7 @@ from transformers import AutoTokenizer, AutoModelForMaskedLM
|
|
| 32 |
model_id = "wallacelw/ModBERTBr"
|
| 33 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 34 |
model = AutoModelForMaskedLM.from_pretrained(model_id)
|
| 35 |
-
text = "
|
| 36 |
inputs = tokenizer(text, return_tensors="pt")
|
| 37 |
outputs = model(**inputs)
|
| 38 |
# To get predictions for the mask:
|
|
@@ -53,7 +53,7 @@ pipe = pipeline(
|
|
| 53 |
"fill-mask",
|
| 54 |
model="wallacelw/ModBERTBr",
|
| 55 |
)
|
| 56 |
-
input_text = "
|
| 57 |
results = pipe(input_text)
|
| 58 |
pprint(results)
|
| 59 |
```
|
|
|
|
| 32 |
model_id = "wallacelw/ModBERTBr"
|
| 33 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 34 |
model = AutoModelForMaskedLM.from_pretrained(model_id)
|
| 35 |
+
text = "A capital do Brasil é [MASK]."
|
| 36 |
inputs = tokenizer(text, return_tensors="pt")
|
| 37 |
outputs = model(**inputs)
|
| 38 |
# To get predictions for the mask:
|
|
|
|
| 53 |
"fill-mask",
|
| 54 |
model="wallacelw/ModBERTBr",
|
| 55 |
)
|
| 56 |
+
input_text = "O planeta [MASK] é o terceiro do sistema solar."
|
| 57 |
results = pipe(input_text)
|
| 58 |
pprint(results)
|
| 59 |
```
|