update usage example
Browse files
README.md
CHANGED
@@ -34,7 +34,7 @@ with torch.no_grad():
|
|
34 |
probs = model(**input).logits
|
35 |
|
36 |
predicted_label = torch.argmax(probs, dim=1).item()
|
37 |
-
print(predicted_label)
|
38 |
```
|
39 |
|
40 |
## Citation
|
|
|
34 |
probs = model(**input).logits
|
35 |
|
36 |
predicted_label = torch.argmax(probs, dim=1).item()
|
37 |
+
print(model.config.id2label[predicted_label])
|
38 |
```
|
39 |
|
40 |
## Citation
|