Update README.md

README.md

@@ -54,6 +54,7 @@ pip install transformers torch
 
 ```bash
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
+import torch
 
 labels = [
     'transaction_query',
@@ -63,6 +64,7 @@ labels = [
     'credi_card_application',
     'balance_inquiry'
 ]
+label2id = {label: idx for idx, label in enumerate(labels)}
 id2label = {idx: label for label, idx in label2id.items()}
 
 # Load the pre-trained model and tokenizer