Update README.md

README.md

@@ -186,8 +186,11 @@ predictions, logit_output = model.predict(["How to create a good text classifica
                                         "On our site, you can find a great genre identification model which you can use for thousands of different tasks. With our model, you can fastly and reliably obtain high-quality genre predictions and explore which genres exist in your corpora. Available for free!"]
                                         )
 predictions
-### Output:
-### (['Instruction', 'Promotion']
+# Output: array([3, 8])
+
+[model.config.id2label[i] for i in predictions]
+# Output: ['Instruction', 'Promotion']
+
 ```
 
 ## Performance