Update README.md

README.md

@@ -36,7 +36,20 @@ Current Labels Available:
 
 ```python
 from classifier import MistralForSequenceClassification
+from transformers import AutoTokenizer
+from transformers import pipeline
+
+
 model = MistralForSequenceClassification.from_pretrained('malaysia-ai/malaysian-sfw-classifier')
+tokenizer = AutoTokenizer.from_pretrained('malaysia-ai/malaysian-sfw-classifier')
+
+
+pipe = pipeline("text-classification",
+                 tokenizer = tokenizer,
+                 model=model)
+
+input_str = ["INSERT_INPUT_HERE"]
+print(pipe(input_str))
 ```