Update README.md

README.md

@@ -68,7 +68,7 @@ The model's performance is dependent on the nature and quality of the training d
 ### Transformers
 
 ```python
-from transformers import AutoTokenizer, AutoModelForSequenceClassification
+from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 import torch
 
 tokenizer = AutoTokenizer.from_pretrained("laiyer/deberta-v3-base-prompt-injection")
@@ -80,7 +80,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("Your prompt injection is here"))