Commit
•
ac139a9
1
Parent(s):
3fc7f21
Update README.md
Browse files
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=
|
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])
|