Update tasks/text.py

tasks/text.py

@@ -100,8 +100,8 @@ 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(text_input_ids, text_attention_mask)
-        predictions = torch.argmax(outputs.logits, dim=1).cpu().numpy()
+        logits = model(text_input_ids, text_attention_mask)
+        predictions = torch.argmax(logits, dim=1).cpu().numpy()
     
     #--------------------------------------------------------------------------------------------
     # YOUR MODEL INFERENCE STOPS HERE