Update README.md

README.md

@@ -13,3 +13,8 @@ widget:
 
 - I have used IMDB movie review dataset to create custom model by using DistilBertForSequenceClassification.
 
+from transformers import DistilBertForSequenceClassification, Trainer, TrainingArguments
+
+model = DistilBertForSequenceClassification.from_pretrained("Sakil/imdbsentdistilbertmodel")
+tokenizer = tokenizer.from_pretrained("Sakil/imdbsentdistilbertmodel")
+