israelcamp commited on
Commit
519c5f0
1 Parent(s): 7314c2a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +33 -0
README.md CHANGED
@@ -1,3 +1,36 @@
1
  ---
 
 
 
 
2
  license: mit
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ inference: false
3
+ language: pt
4
+ datasets:
5
+ - lener_br
6
  license: mit
7
  ---
8
+
9
+ # DeBERTinha XSmall for NER
10
+
11
+
12
+ ## Full classification example
13
+
14
+ ```python
15
+ from transformers import AutoModelForSequenceClassification, AutoTokenizer, AutoConfig
16
+ import torch
17
+
18
+ model_name = "sagui-nlp/debertinha-ptbr-xsmall-lenerbr"
19
+ model = AutoModelForSequenceClassification.from_pretrained(model_name)
20
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
21
+
22
+ input_text = "Acrescento que não há de se falar em violação do artigo 114, § 3º, da Constituição Federal, posto que referido dispositivo revela-se impertinente, tratando da possibilidade de ajuizamento de dissídio coletivo pelo Ministério Público do Trabalho nos casos de greve em atividade essencial."
23
+
24
+ inputs = tokenizer(input_text, max_length=512, truncation=True, return_tensors="pt")
25
+ tokens = inputs.tokens()
26
+
27
+ outputs = model(**inputs).logits
28
+ predictions = torch.argmax(outputs, dim=2)
29
+
30
+ for token, prediction in zip(tokens, predictions[0].numpy()):
31
+ print((token, model.config.id2label[prediction]))
32
+ ```
33
+
34
+ ## Citation
35
+
36
+ Comming soon