Update ner.py

ner.py

@@ -2,8 +2,8 @@ from transformers import AutoTokenizer, AutoModelForTokenClassification
 from transformers import pipeline
 from collections import defaultdict
 import torch
-#device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-device = torch.device("cuda")
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+#device = torch.device("cuda")
 tokenizer = AutoTokenizer.from_pretrained("Leo97/KoELECTRA-small-v3-modu-ner")
 model = AutoModelForTokenClassification.from_pretrained("Leo97/KoELECTRA-small-v3-modu-ner")
 model.to(device)