Logs

app.py

@@ -126,7 +126,7 @@ def evaluate_model(model, dataloader, device):
         for batch in dataloader:
             input_ids = batch['input_ids'].to(device)
             attention_mask = batch['attention_mask'].to(device)
-            labels = batch['labels'].to(device)
+            labels = batch['labels'].to(device).cpu().numpy()
             x = len(labels[0])
             print(labels[0])
             for l in labels: