Text Classification
Transformers
TensorBoard
Safetensors
modernbert
wissamantoun commited on
Commit
83383cc
·
verified ·
1 Parent(s): e71daea

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
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