lrds-code commited on
Commit
b27ba23
1 Parent(s): 40c7e38

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +20 -1
README.md CHANGED
@@ -58,4 +58,23 @@ Sym-Pair é composto por:
58
 
59
  ## Como Usar
60
 
61
- TODO
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
58
 
59
  ## Como Usar
60
 
61
+ Exemplo de uma única classificação:
62
+ ​​
63
+ ```python
64
+ import torch
65
+ from transformers import AutoTokenizer
66
+ from transformers import AutoModelForSequenceClassification
67
+
68
+ model_name = 'lrds-code/simnonym'
69
+
70
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
71
+ model = AutoModelForSequenceClassification.from_pretrained(model_name)
72
+
73
+ inputs = tokenizer('As palavras "feliz" e "alegre" são sinônimos?', return_tensors='pt')
74
+
75
+ with torch.no_grad():
76
+ output = model(**inputs).logits
77
+
78
+ predict_id = logits.argmax().item()
79
+ model.config.id2label[predict_id]
80
+ ```