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 |
+
```
|