Update README.md

README.md

@@ -85,7 +85,8 @@ The inference code for this model is available through the NeMo Curator GitHub r
 
 # How to Use in Transformers
 To use the multilingual domain classifier, use the following code:
-```
+
+```python
 import torch
 from torch import nn
 from transformers import AutoModel, AutoTokenizer, AutoConfig
@@ -118,6 +119,7 @@ outputs = model(inputs["input_ids"], inputs["attention_mask"])
 predicted_classes = torch.argmax(outputs, dim=1)
 predicted_domains = [config.id2label[class_idx.item()] for class_idx in predicted_classes.cpu().numpy()]
 print(predicted_domains)
+# ['Sports', 'News']
 ```
 
 # Input & Output