Logs

app.py

@@ -164,8 +164,8 @@ def evaluate_model(model, dataloader, device):
                 if sample_count < num_samples:
                     print(f"Sample {sample_count + 1}:")
                     print(f"Tokens: {tokenizer.convert_ids_to_tokens(input_ids[i])}")
-                    print(f"True Labels: {[id2label[label] for label in valid_labels]}")
-                    print(f"Predicted Labels: {[id2label[pred] for pred in valid_preds]}")
+                    print(f"True Labels: {[id2label[label.item()] for label in valid_labels]}")
+                    print(f"Predicted Labels: {[id2label[pred.item()] for pred in valid_preds]}")
                     print("-" * 50)
                     sample_count += 1
                 all_preds.extend(valid_preds.tolist())