diff --git a/inference.py b/inference.py index 7117c1106731499f0d9480be4fe8d87a0925299e..b85f07bcdc4d6d27d8d37793adecafc1f5aba885 100644 --- a/inference.py +++ b/inference.py @@ -99,8 +99,8 @@ for batch in train_dataloader: prediction = torch.argmax(outputs[0]) print(prediction) if prediction == 1: - print("metonymy") + print("Metonymy! The target word has metonymic meaning in this context.") elif prediction == 0: - print("literal") + print("Literal! The target word has literal meaning in this context.")