Update README.md

README.md

@@ -51,8 +51,12 @@ pip install bert-for-sequence classfication
 
 ```python
 from bert_clf import EncoderCLF
+import torch
 
 model = EncoderCLF("whitefoxredhell/language_identification")
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model = model.to(device)
+model = model.eval()
 
 text = "London is the capital of Great Britain"