Update README.md

README.md

@@ -54,11 +54,16 @@ Use the code below to get started with the model.
 
 ```
 from transformers import AutoModelForTokenClassification, AutoTokenizer
+from transformers import pipeline
 
 model = AutoModelForTokenClassification.from_pretrained("rv2307/electra-small-ner")
 tokenizer = AutoTokenizer.from_pretrained("rv2307/electra-small-ner")
 
-nlp = pipeline("ner", model=model, tokenizer=tokenizer,device=device,aggregation_strategy = "max")
+nlp = pipeline("ner",
+              model=model,
+              tokenizer=tokenizer,device="cpu",
+              aggregation_strategy = "max")
+
 ```
 
 ## Training Details