Commit
•
7567e9f
1
Parent(s):
a7498b6
Update README.md
Browse files
README.md
CHANGED
@@ -49,7 +49,7 @@ tokens = tokenizer(["String to be analized"], return_tensors="pt",
|
|
49 |
padding=True, truncation=True, max_length=512)
|
50 |
lgpd_pii_identifier_outputs = lgpd_pii_identifier(**tokens)
|
51 |
|
52 |
-
preds = [pred_mapper[np.argmax(pred)] for pred in
|
53 |
```
|
54 |
## Author
|
55 |
|
|
|
49 |
padding=True, truncation=True, max_length=512)
|
50 |
lgpd_pii_identifier_outputs = lgpd_pii_identifier(**tokens)
|
51 |
|
52 |
+
preds = [pred_mapper[np.argmax(pred)] for pred in lgpd_pii_identifier_outputs.logits.cpu().detach().numpy()]
|
53 |
```
|
54 |
## Author
|
55 |
|