Fix import and typo in device

- pipeline wasn't imported so it was giving error.
- It should be "cpu" instead of "CPU" in ``device=torch.device("cuda" if torch.cuda.is_available() else "CPU"),``

README.md

@@ -59,7 +59,7 @@ Additionally, `distilroberta-base` is case-sensitive model.
 ### Transformers
 
 ```python
-from transformers import AutoTokenizer, AutoModelForSequenceClassification
+from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 import torch
 
 tokenizer = AutoTokenizer.from_pretrained("ProtectAI/distilroberta-base-rejection-v1")
@@ -71,7 +71,7 @@ classifier = pipeline(
   tokenizer=tokenizer,
   truncation=True,
   max_length=512,
-  device=torch.device("cuda" if torch.cuda.is_available() else "CPU"),
+  device=torch.device("cuda" if torch.cuda.is_available() else "cpu"),
 )
 
 print(classifier("Sorry, but I can't assist with that."))