preds eval to numpy

qwen_classifier/evaluate.py

@@ -146,9 +146,8 @@ def _evaluate_local(test_data_path, hf_repo):
 
             logits = global_model(batch["input_ids"], batch["attention_mask"])
 
-            preds = torch.sigmoid(logits).cpu() > 0.5  # Keeps as PyTorch tensor
-            preds = preds.float()  # Convert to 0.0/1.0 if needed
-            labels = labels.cpu()
+            preds = torch.sigmoid(logits).cpu().numpy() > 0.5
+            labels = labels.cpu().numpy()
 
             all_preds.extend(preds)
             all_labels.extend(labels)