Update README.md

README.md

@@ -45,7 +45,7 @@ model = AutoModelForMaskedLM.from_pretrained("ltg/norbert3-base", trust_remote_c
 
 mask_id = tokenizer.convert_tokens_to_ids("[MASK]")
 input_text = tokenizer("Nå ønsker de seg en[MASK] bolig.", return_tensors="pt")
-output_p = bert(**input_text)
+output_p = model(**input_text)
 output_text = torch.where(input_text.input_ids == mask_id, output_p.logits.argmax(-1), input_text.input_ids)
 
 # should output: '[CLS] Nå ønsker de seg en ny bolig.[SEP]'