Update README.md

README.md

@@ -17,6 +17,12 @@ LABEL_1 - Helpful
 The following code shows how to make a prediction with this model
 
 ```python
+from transformers import (
+    AutoTokenizer,
+    AutoModelForSequenceClassification,
+    TextClassificationPipeline,
+)
+
 tokenizer = AutoTokenizer.from_pretrained("banjtheman/distilbert-base-uncased-helpful-amazon")
 
 model = AutoModelForSequenceClassification.from_pretrained(