wallacelw commited on
Commit
3b571fb
·
verified ·
1 Parent(s): 61c6d57

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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 = "The capital of France is [MASK]."
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 = "He walked to the [MASK]."
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
  ```