Update README.md

README.md

@@ -9,7 +9,15 @@ language:
   </summary>
 
 ### Fetching the model
+
 ```python
+import torch
+from torch.utils.data import DataLoader, Dataset
+from transformers import AutoTokenizer, AutoModelForSequenceClassification, AdamW
+from sklearn.model_selection import train_test_split
+import pandas as pd
+from tqdm import tqdm
+
 # Load the TinyBERT tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained('huawei-noah/TinyBERT_General_4L_312D')
 model = AutoModelForSequenceClassification.from_pretrained('huawei-noah/TinyBERT_General_4L_312D', num_labels=2)