UlrikKoren
/

PIIMask-EN

Model card Files Files and versions Community

UlrikKoren commited on 16 days ago

Commit

3d41b9a

•

1 Parent(s): e38593a

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -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)
 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])

 ]
 inputs = tokenizer.apply_chat_template(chat, tokenize=False, return_tensors="pt", padding=True, truncation=False)
+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])