Update tasks/text.py

tasks/text.py

@@ -107,8 +107,10 @@ async def evaluate_text(request: TextEvaluationRequest):
     test_dataloader = DataLoader(test_data, sampler=test_sampler, batch_size=batch_size)
 
     predictions = []
+    c=0
     for batch in tqdm(test_dataloader):
-
+        print(c)
+        c+=1
 
         b_input_ids, b_input_mask, b_token_type_ids = batch
         with torch.no_grad():
@@ -116,7 +118,7 @@ async def evaluate_text(request: TextEvaluationRequest):
 
         logits = logits.detach().cpu().numpy()
         predictions.extend(logits.argmax(1))
-
+        
     
     true_labels = test_dataset["label"]   
     # Make random predictions (placeholder for actual model inference)