UlrikKoren commited on
Commit
ac139a9
1 Parent(s): 3fc7f21

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -72,7 +72,7 @@ chat = [
72
  ]
73
 
74
  inputs = tokenizer.apply_chat_template(chat, tokenize=False, return_tensors="pt", padding=True, truncation=False)
75
- model = load_model("UlrikKoren/PIIMask-NOR", step=1)
76
  outputs = model.generate(input_ids=inputs['input_ids'].to(device), max_new_tokens=2048)
77
  decoded_outputs = [tokenizer.decode(output, skip_special_tokens=False) for output in outputs]
78
  print(decoded_outputs[0])
 
72
  ]
73
 
74
  inputs = tokenizer.apply_chat_template(chat, tokenize=False, return_tensors="pt", padding=True, truncation=False)
75
+ model = load_model("UlrikKoren/PIIMask-NOR", step=1159)
76
  outputs = model.generate(input_ids=inputs['input_ids'].to(device), max_new_tokens=2048)
77
  decoded_outputs = [tokenizer.decode(output, skip_special_tokens=False) for output in outputs]
78
  print(decoded_outputs[0])