rafech commited on
Commit
21d957b
·
verified ·
1 Parent(s): b5bdf99

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. tasks/text.py +1 -1
tasks/text.py CHANGED
@@ -93,7 +93,7 @@ async def evaluate_text(request: TextEvaluationRequest):
93
  outputs = model(input_ids, attention_mask=attention_mask)
94
  preds = torch.argmax(outputs.logits, dim=1)
95
  predictions.extend(preds.cpu().numpy())
96
- print('here's a batch')
97
 
98
  true_labels = test_dataset["label"]
99
 
 
93
  outputs = model(input_ids, attention_mask=attention_mask)
94
  preds = torch.argmax(outputs.logits, dim=1)
95
  predictions.extend(preds.cpu().numpy())
96
+ print('here is a batch')
97
 
98
  true_labels = test_dataset["label"]
99