Update ccl_win.py

ccl_win.py

@@ -132,8 +132,8 @@ class Scorer:
         scores = []
         n_preds = len(pred_list)
         with torch.no_grad():
-            for b in range(0,n_preds,batch_size):
-                strides = [x.lower() for x in pred_list[b:b+batch_size]]
+            for b in range(0,n_preds,self.batch_size):
+                strides = [x.lower() for x in pred_list[b:b+self.batch_size]]
                 tinput = self._tokenizer(strides,padding=True,truncation=True,max_length=512,return_tensors="pt")
                 tinput = {k:v.to(self.device) for k,v in tinput.items()}
                 output = self._model(**tinput)