FelipeCasali-USP commited on
Commit
7567e9f
1 Parent(s): a7498b6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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 finbertptbr_outputs.logits.cpu().detach().numpy()]
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