Update README.md
Browse files
README.md
CHANGED
@@ -57,4 +57,4 @@ inputs = tokenizer(model_prompt, return_tensors="pt").to(device)
|
|
57 |
# adjust max_new_tokens according to your need
|
58 |
outputs = pii_model.generate(**inputs, do_sample=True, max_new_tokens=120)
|
59 |
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
60 |
-
print(response)
|
|
|
57 |
# adjust max_new_tokens according to your need
|
58 |
outputs = pii_model.generate(**inputs, do_sample=True, max_new_tokens=120)
|
59 |
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
60 |
+
print(response) # {'firstname': ['Abner'], 'dob': ['23/03/1926'], 'email': ['[email protected]']}
|