Update tasks/text.py

tasks/text.py

@@ -99,7 +99,7 @@ async def evaluate_text(request: TextEvaluationRequest):
     with torch.no_grad():
         text_input_ids = text_encoding["input_ids"].to(device)
         text_attention_mask = text_encoding["attention_mask"].to(device)
-        outputs = model(test_input_ids, test_attention_mask)
+        outputs = model(text_input_ids, text_attention_mask)
         predictions = torch.argmax(outputs.logits, dim=1).cpu().numpy()
 
     #--------------------------------------------------------------------------------------------