Update README.md

README.md

@@ -18,7 +18,6 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
 id2label = {0: "negative", 1: "neutral", 2: "positive"}
 tokenizer = AutoTokenizer.from_pretrained("Voicelab/herbert-base-cased-sentiment")
-
 model = AutoModelForSequenceClassification.from_pretrained("Voicelab/herbert-base-cased-sentiment")
 
 input = ["Ale fajnie, spadł dzisiaj śnieg! Ulepimy dziś bałwana?"]
@@ -38,6 +37,10 @@ print(input, "--->", prediction)
 
 ```
 
+Predicted output:
+```python
+['Ale fajnie, spadł dzisiaj śnieg! Ulepimy dziś bałwana?'] ---> positive
+```
 
 ### Overview
 - **Language model:** [allegro/herbert-base-cased](https://huggingface.co/allegro/herbert-base-cased)