Update README.md

README.md

@@ -11,13 +11,6 @@ model-index:
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
-label_mapping = {
-    'normal': 0,
-    'ddos': 1,
-    'injection': 2,
-    'password': 3,
-    'scanning': 4
-}
 
 # ton_iot_attempt
 
@@ -25,6 +18,14 @@ This model is a fine-tuned version of [roberta-large](https://huggingface.co/rob
 It achieves the following results on the evaluation set:
 - Loss: 0.0380
 
+label_mapping = {
+  'normal': 0,
+  'ddos': 1,
+  'injection': 2,
+  'password': 3,
+  'scanning': 4
+}
+
 ## Model description
 
 More information needed