Update README.md

README.md

@@ -73,7 +73,7 @@ chat = [
 ]
 
 inputs = tokenizer.apply_chat_template(chat, tokenize=False, return_tensors="pt", padding=True, truncation=False)
-model = load_model("UlrikKoren/PIIMask-EN", step=1)
+model = load_model("UlrikKoren/PIIMask-EN", step=1159)
 outputs = model.generate(input_ids=inputs['input_ids'].to(device), max_new_tokens=2048)
 decoded_outputs = [tokenizer.decode(output, skip_special_tokens=False) for output in outputs]
 print(decoded_outputs[0])