Update README.md

README.md

@@ -15,11 +15,13 @@ NUM_LABELS= len(labels)
 id2label={id:label for id,label in enumerate(labels)}
 
 label2id={label:id for id,label in enumerate(labels)}
+```
 
 ```
 tokenizer = BertTokenizerFast.from_pretrained("bert-base-uncased", max_length=512)
 model = BertForSequenceClassification.from_pretrained("bert-base-uncased", num_labels=NUM_LABELS, id2label=id2label, label2id=label2id)
 
+```
 ```
 def predict(text):
     """