Update tasks/text.py

tasks/text.py

@@ -82,11 +82,11 @@ async def evaluate_text(request: TextEvaluationRequest):
     
     predictions = []
     with torch.no_grad():
-    for batch in test_loader:
-        input_ids, attention_mask, labels = [x.to(device) for x in batch]
-        outputs = model(input_ids, attention_mask=attention_mask)
-        preds = torch.argmax(outputs.logits, dim=1)
-        predictions.extend(preds.cpu().numpy())
+        for batch in test_loader:
+            input_ids, attention_mask, labels = [x.to(device) for x in batch]
+            outputs = model(input_ids, attention_mask=attention_mask)
+            preds = torch.argmax(outputs.logits, dim=1)
+            predictions.extend(preds.cpu().numpy())
 
     true_labels = test_dataset["label"]