Update README.md
Browse files
README.md
CHANGED
@@ -58,4 +58,23 @@ Sym-Pair é composto por:
|
|
58 |
|
59 |
## Como Usar
|
60 |
|
61 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 |
+
```
|