Logs

app.py

@@ -147,7 +147,7 @@ def evaluate_model(model, dataloader, device):
 
             # Get predictions
             preds = torch.argmax(logits, dim=-1).cpu().numpy()
-            mask = attention_mask.cpu().numpy().astype(bool)
+            mask = attention_mask
             
             # Process each sequence in the batch
             for i in range(current_batch_size):