Update README.md
Browse files
README.md
CHANGED
|
@@ -83,7 +83,7 @@ with torch.no_grad():
|
|
| 83 |
probs = torch.softmax(model(**inputs).logits, dim=-1)[0]
|
| 84 |
|
| 85 |
print({model.config.id2label[i]: round(p.item(), 3) for i, p in enumerate(probs)})
|
| 86 |
-
|
| 87 |
|
| 88 |
Labels
|
| 89 |
id label
|
|
|
|
| 83 |
probs = torch.softmax(model(**inputs).logits, dim=-1)[0]
|
| 84 |
|
| 85 |
print({model.config.id2label[i]: round(p.item(), 3) for i, p in enumerate(probs)})
|
| 86 |
+
```
|
| 87 |
|
| 88 |
Labels
|
| 89 |
id label
|