Update tasks/text.py

tasks/text.py

@@ -84,6 +84,7 @@ async def evaluate_text(request: TextEvaluationRequest):
     
     test_dataset_0 = TensorDataset(test_encodings["input_ids"], test_encodings["attention_mask"], test_labels)
     test_loader = DataLoader(test_dataset_0, batch_size=16)
+    print('encoded')
     
     predictions = []
     with torch.no_grad():
@@ -92,6 +93,7 @@ async def evaluate_text(request: TextEvaluationRequest):
             outputs = model(input_ids, attention_mask=attention_mask)
             preds = torch.argmax(outputs.logits, dim=1)
             predictions.extend(preds.cpu().numpy())
+            print('here's a batch')
 
     true_labels = test_dataset["label"]