Update tasks/text.py

tasks/text.py

@@ -159,9 +159,8 @@ async def evaluate_text(request: TextEvaluationRequest):
     predictions = []
     for batch in tqdm(test_dataloader):
 
-        b_input_ids, b_input_mask, b_token_type_ids = batch
         with torch.no_grad():
-            logits = model(b_input_ids, b_token_type_ids, b_input_mask)    
+            logits = model(**batch)    
 
         logits = logits.detach().cpu().numpy()
         predictions.extend(logits.argmax(1))