josu commited on
Commit
1c5edc9
1 Parent(s): 75c3b4e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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
- unmasker1 = pipeline('fill-mask', model=model, tokenizer=tokenizer ,device=0)
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)