Logs

app.py

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