Update README.md
Browse files
README.md
CHANGED
@@ -12,7 +12,7 @@ widget:
|
|
12 |
from transformers import pipeline, AlbertTokenizer, AlbertForMaskedLM
|
13 |
model = AlbertForMaskedLM.from_pretrained('josu/albert-pt-br')
|
14 |
tokenizer = AlbertTokenizer.from_pretrained('josu/albert-pt-br')
|
15 |
-
|
16 |
|
17 |
text = 'Marte está no [MASK] solar.'
|
18 |
unmasker(text)
|
|
|
12 |
from transformers import pipeline, AlbertTokenizer, AlbertForMaskedLM
|
13 |
model = AlbertForMaskedLM.from_pretrained('josu/albert-pt-br')
|
14 |
tokenizer = AlbertTokenizer.from_pretrained('josu/albert-pt-br')
|
15 |
+
unmasker = pipeline('fill-mask', model=model, tokenizer=tokenizer ,device=0)
|
16 |
|
17 |
text = 'Marte está no [MASK] solar.'
|
18 |
unmasker(text)
|