predictions = model.predict(x_test[:3])
predicted_labels = np.argmax(predictions, axis=1)

print("Predicted label:", predicted_labels)
print("True label:", y_test[:3])

        __
        __