Update tamilatis/predict.py

tamilatis/predict.py

@@ -18,7 +18,7 @@ class TamilATISPredictor:
         self.model = model
         self.model.eval()
         self.tokenizer = AutoTokenizer.from_pretrained(tokenizer)
-        self.device = "cuda" if torch.cuda.is_available() else "CPU"
+        self.device = "cuda" if torch.cuda.is_available() else "cpu"
         self.model.load_state_dict(torch.load(checkpoint_path,map_location=self.device))
         self.num_labels = num_labels
         self.label_encoder = label_encoder