Update README.md
Browse files
README.md
CHANGED
@@ -54,6 +54,7 @@ print(probs.argmax(dim=-1))
|
|
54 |
```
|
55 |
|
56 |
You can convert the `logits` of the model with a softmax to obtain a probability distribution over the following 24 categories (in order of labels, also see `id2label` and `label2id` in the model config):
|
|
|
57 |
0. Academic Writing
|
58 |
1. Content Listing
|
59 |
2. Creative Writing
|
|
|
54 |
```
|
55 |
|
56 |
You can convert the `logits` of the model with a softmax to obtain a probability distribution over the following 24 categories (in order of labels, also see `id2label` and `label2id` in the model config):
|
57 |
+
|
58 |
0. Academic Writing
|
59 |
1. Content Listing
|
60 |
2. Creative Writing
|