Fix import and typo in device (#1)

- Fix import and typo in device (3df0acb1b8d1b5015b2882b623b7503530e95172)


Co-authored-by: rohinish <[email protected]>

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."))