Change print

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.")