Update README.md

README.md

@@ -5,14 +5,14 @@ tags: []
 
 ## Inference
 ```python
-from transformers import AutoModelForSequenceClassification, DistilBertTokenizer
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import time
 import torch
 import re
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 model = AutoModelForSequenceClassification.from_pretrained("Mr-Vicky-01/finetuned_classification_model").to(device)
-tokenizer = DistilBertTokenizer.from_pretrained("Mr-Vicky-01/finetuned_classification_model")
+tokenizer = AutoTokenizer.from_pretrained("Mr-Vicky-01/finetuned_classification_model")
 
 
 start = time.time()