davda54 commited on
Commit
316c124
1 Parent(s): 0811758

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -31,8 +31,8 @@ import torch
31
  from transformers import AutoTokenizer
32
  from modeling_norbert import NorbertForMaskedLM
33
 
34
- tokenizer = AutoTokenizer.from_pretrained(path/to/folder)
35
- bert = NorbertForMaskedLM.from_pretrained(path/to/folder)
36
 
37
  mask_id = tokenizer.convert_tokens_to_ids("[MASK]")
38
  input_text = tokenizer("Nå ønsker de seg en[MASK] bolig.", return_tensors="pt")
 
31
  from transformers import AutoTokenizer
32
  from modeling_norbert import NorbertForMaskedLM
33
 
34
+ tokenizer = AutoTokenizer.from_pretrained("path/to/folder")
35
+ bert = NorbertForMaskedLM.from_pretrained("path/to/folder")
36
 
37
  mask_id = tokenizer.convert_tokens_to_ids("[MASK]")
38
  input_text = tokenizer("Nå ønsker de seg en[MASK] bolig.", return_tensors="pt")