Updating stylization

README.md

@@ -11,19 +11,6 @@ A tiny comment toxicity classifier model at only 2M parameters. With only ~10MB
 A paper on this model is being released soon.
 
 
-## Benchmarks
-
-The Tiny-Toxic-Detector achieves an impressive 90.26% on the Toxigen benchmark and 87.34% on the Jigsaw-Toxic-Comment-Classification-Challenge. Here we compare our results against other toxic classification models:
-
-
-| Model                             | Size (parameters) | Toxigen (%) | Jigsaw (%) |
-| --------------------------------- | ----------------- | ----------- | ---------- |
-| lmsys/toxicchat-t5-large-v1.0     | 738M              | 72.67       | 88.82      |
-| s-nlp/roberta toxicity classifier | 124M              | 88.41       | **94.92**  |
-| mohsenfayyaz/toxicity-classifier  | 109M              | 81.50       | 83.31      |
-| martin-ha/toxic-comment-model     | 67M               | 68.02       | 91.56      |
-| **Tiny-toxic-detector**           | **2M**            | **90.97**   | 86.98      |
-
 
 ## Usage
 This model uses custom architecture and requires some extra custom code to work. Below you can find the architecture and a fully-usable example.
@@ -204,6 +191,20 @@ with torch.no_grad():
   prediction = "Toxic" if logits > 0.5 else "Not Toxic"
 ```
 
+## Benchmarks
+
+The Tiny-Toxic-Detector achieves an impressive 90.26% on the Toxigen benchmark and 87.34% on the Jigsaw-Toxic-Comment-Classification-Challenge. Here we compare our results against other toxic classification models:
+
+
+| Model                             | Size (parameters) | Toxigen (%) | Jigsaw (%) |
+| --------------------------------- | ----------------- | ----------- | ---------- |
+| lmsys/toxicchat-t5-large-v1.0     | 738M              | 72.67       | 88.82      |
+| s-nlp/roberta toxicity classifier | 124M              | 88.41       | **94.92**  |
+| mohsenfayyaz/toxicity-classifier  | 109M              | 81.50       | 83.31      |
+| martin-ha/toxic-comment-model     | 67M               | 68.02       | 91.56      |
+| **Tiny-toxic-detector**           | **2M**            | **90.97**   | 86.98      |
+
+
 
 ## Usage and Limitations